Use integrations only in master process (#89)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
single_channel
Ayush Chaurasia 2 years ago committed by GitHub
parent cb4f20f3cf
commit 5fbea25f0b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -86,6 +86,7 @@ class BaseTrainer:
for callback, func in callbacks.default_callbacks.items(): for callback, func in callbacks.default_callbacks.items():
self.add_callback(callback, func) self.add_callback(callback, func)
if RANK in {0, -1}:
callbacks.add_integration_callbacks(self) callbacks.add_integration_callbacks(self)
def add_callback(self, onevent: str, callback): def add_callback(self, onevent: str, callback):

Loading…
Cancel
Save