Skip to content

Conversation

mayank31398
Copy link
Collaborator

No description provided.

Comment on lines -56 to +59
torch.cuda.empty_cache()
gc.collect()
torch.cuda.synchronize()
if args.clear_every_run:
torch.cuda.empty_cache()
gc.collect()
torch.cuda.synchronize()
Copy link
Collaborator Author

Choose a reason for hiding this comment

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

this is the fix for ds-inference

mayank31398 and others added 9 commits December 5, 2022 19:24
* input length experiments * sort input lengths in ascending order * make default max input length -1 make some updates to Alex's code
@mayank31398 mayank31398 merged commit 2478982 into main Dec 6, 2022
@mayank31398 mayank31398 deleted the mayank/refactor branch December 6, 2022 21:54
@mayank31398
Copy link
Collaborator Author

Thanks Alex

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

2 participants