Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Attention with scaling #115

Merged
merged 103 commits into from
Jul 30, 2024
Merged

Attention with scaling #115

merged 103 commits into from
Jul 30, 2024

Conversation

larascipio
Copy link
Contributor

Changes:

  • In selection.py I created one single function that handles all steps of the process (instead of two separate functions)
  • In attention module I changed the corruption functions to one-sided or two-sided corruption, and I also added some variables that I used for testing (e.g. 'test_sides_corruption'). These might be unnecessary for you.

Copy link
Collaborator

@bdvllrs bdvllrs left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Mypy is not passing... Will investigate before merging

@bdvllrs bdvllrs force-pushed the attention-with-scaling branch from bc3ba39 to a5b26a8 Compare July 30, 2024 12:48
@bdvllrs bdvllrs merged commit 774ff2f into main Jul 30, 2024
1 check passed
@bdvllrs bdvllrs deleted the attention-with-scaling branch July 30, 2024 13:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants