From 5fbea25f0b3b26b0d6883ca53eaf6a2447f633ad Mon Sep 17 00:00:00 2001 From: Ayush Chaurasia Date: Sat, 24 Dec 2022 22:09:30 +0530 Subject: [PATCH] Use integrations only in master process (#89) Co-authored-by: Glenn Jocher --- ultralytics/yolo/engine/trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ultralytics/yolo/engine/trainer.py b/ultralytics/yolo/engine/trainer.py index bcb64f5..ac9770a 100644 --- a/ultralytics/yolo/engine/trainer.py +++ b/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): """