Training Improvements, LR Scheduler, GPU Fixes, and PyPI Package#93
Open
JinruG wants to merge 8 commits intoGhadjeres:masterfrom
Open
Training Improvements, LR Scheduler, GPU Fixes, and PyPI Package#93JinruG wants to merge 8 commits intoGhadjeres:masterfrom
JinruG wants to merge 8 commits intoGhadjeres:masterfrom
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Training Improvements
cuda()called beforetrain(): The model previously was moved to GPU only after training, causing the entire training loop to run on CPU.--lr_patienceand--lr_factoroptions above.Dependency Compatibility
torch.load(...)bare calls with explicitweights_only=True/weights_only=Falsekeyword to silence theFutureWarningintroduced in PyTorch 2.0 and enforced as an error in PyTorch 2.4+.standalone_modein the__main__block is now set toFalseso that Click returns the result value rather than callingsys.exit(0)on success — required for embedding the entry point in notebooks or test harnesses without process termination.deepbach_pytorchPyPI Packagehttps://pypi.org/manage/project/deepbach-pytorch/release/0.3.6/
A test script (
test.py) is included that exercises the public API entry point:test.py