-
Notifications
You must be signed in to change notification settings - Fork 1
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
Deterministic Training Support #23
Conversation
Test Results 6 files 6 suites 38m 41s ⏱️ For more details on these failures and errors, see this check. Results for commit c5b77be. ♻️ This comment has been updated with latest results. |
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.
Looks good, thanks!
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.
LGTM
* Added seed to config for reproducibility * fixed seg drawing when using torch deterministic backend * added deterministic order of creating nodes * removed seed from example config * added reproducability to inspect * formatting --------- Co-authored-by: DrejcPesjak <[email protected]> Co-authored-by: Martin Kozlovský <[email protected]>
Added option for full reproducibility of training by specifying the seed.
This also includes deterministic order of creating
Nodes
and using CPU for drawing segmentation masks for visualizations (related to this torchvision bug)