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

Deterministic Training Support #23

Merged
merged 7 commits into from
May 15, 2024
Merged

Deterministic Training Support #23

merged 7 commits into from
May 15, 2024

Conversation

klemen1999
Copy link
Collaborator

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)

Copy link

github-actions bot commented May 7, 2024

Test Results

  6 files    6 suites   38m 41s ⏱️
 57 tests  25 ✅ 0 💤  7 ❌  25 🔥
342 runs  178 ✅ 0 💤 14 ❌ 150 🔥

For more details on these failures and errors, see this check.

Results for commit c5b77be.

♻️ This comment has been updated with latest results.

Copy link
Collaborator

@tersekmatija tersekmatija left a comment

Choose a reason for hiding this comment

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

Looks good, thanks!

luxonis_train/core/core.py Show resolved Hide resolved
@tersekmatija tersekmatija requested a review from kozlov721 May 8, 2024 09:22
Copy link
Collaborator

@kozlov721 kozlov721 left a comment

Choose a reason for hiding this comment

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

LGTM

@kozlov721 kozlov721 added the enhancement New feature or request label May 13, 2024
@kozlov721 kozlov721 changed the title Feature/seed Fixed Seed Support May 13, 2024
@kozlov721 kozlov721 changed the title Fixed Seed Support Deterministic Training Support May 15, 2024
@kozlov721 kozlov721 merged commit 5a31f72 into dev May 15, 2024
2 of 8 checks passed
@kozlov721 kozlov721 deleted the feature/seed branch May 15, 2024 18:55
kozlov721 added a commit that referenced this pull request Oct 9, 2024
* 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]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants