-
-
Notifications
You must be signed in to change notification settings - Fork 390
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
Returned resume
support
#1349
Returned resume
support
#1349
Conversation
…ssed from one stage to another)
def to_check_devices(): | ||
cuda_devices = [f"cuda:{i}" for i in range(NUM_CUDA_DEVICES)] | ||
return ["cpu", *cuda_devices] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🤔 should we simplify to cpu
and cuda
only? if it works on one GPU, we suppose, that it will work on the another too ;)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
i reformatted old test, but sure, we can simplify it
Pull request has been modified.
This pull request is now in conflicts. @y-ksenia, could you fix it? 🙏 |
Pull Request FAQ
Description
Returned
resume
to CheckpointCallback; added it's suppert to Runner.train and Runner.predict_loader; updated cli withresume
andautoresume
.In addition changed behaviour of runner.model between stages (now model passes from one stage to another).
Related Issue
#1193
Type of Change
PR review
Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.
Checklist
latest
andminimal
requirements?