-
Notifications
You must be signed in to change notification settings - Fork 240
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
[Torch] Fix classification sample metric dumps not in main process #2146
[Torch] Fix classification sample metric dumps not in main process #2146
Conversation
Codecov Report
Additional details and impacted files@@ Coverage Diff @@
## develop #2146 +/- ##
===========================================
+ Coverage 36.23% 36.24% +0.01%
===========================================
Files 477 477
Lines 42544 42570 +26
===========================================
+ Hits 15414 15428 +14
- Misses 27130 27142 +12 |
aa6a5ba
to
9dd78cf
Compare
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.
Mainly OK, one change to be addressed
if is_main_process(): | ||
if log_validation_info: |
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.
Keep the original version with and
to have less levels of indent
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.
Done
Build is successful, it's safe to merge |
Changes
In torch classification sample:
write_metrics
andmlflow
accuracy logging is called only in main processReason for changes
Torch test_compression_training (build 106) failed due to metric collected by
write_metrics
and metric saved in checkpoint with key "best_acc1" were differentTests
Torch test_compression_training build
109111