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

[P0] remove autograd debugging #98

Merged
merged 1 commit into from
Jan 30, 2024
Merged

[P0] remove autograd debugging #98

merged 1 commit into from
Jan 30, 2024

Conversation

aryamanarora
Copy link
Collaborator

Description

There was an autograd debugging line left in that significantly impacted performance. Discovered it while using a profiler to debug experiment speeds. Removing it reduced runtime for training my DAS benchmark work on pythia-14m from 289s to 170s!

Testing Done

No effect on library behaviours at all, just removed torch.autograd.set_detect_anomaly(True).

Checklist:

  • My PR title strictly follows the format: [Your Priority] Your Title
  • I have attached the testing log above
  • I provide enough comments to my code
  • I have changed documentations
  • I have added tests for my changes

@aryamanarora aryamanarora changed the title remove autograd debugging [P0] remove autograd debugging Jan 30, 2024
@frankaging
Copy link
Collaborator

Thanks for catching this!

@frankaging frankaging merged commit 2531a5f into main Jan 30, 2024
1 check passed
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