Use integrations only in master process (#89)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/90/head
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
  1. 3
      ultralytics/yolo/engine/trainer.py

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

Loading…
Cancel
Save