Comet changes (#9328)

* add link to comte tutorial from main README * fix prediction interval bug
ultralytics · Sep 8, 2022 · 5f075ee · 5f075ee
1 parent 903b239
commit 5f075ee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -157,6 +157,7 @@ python train.py --data coco.yaml --cfg yolov5n.yaml --weights '' --batch-size 12
 - [Roboflow for Datasets, Labeling, and Active Learning](https://github.com/ultralytics/yolov5/issues/4975)  🌟 NEW
 - [ClearML Logging](https://github.com/ultralytics/yolov5/tree/master/utils/loggers/clearml) 🌟 NEW
 - [Deci Platform](https://github.com/ultralytics/yolov5/wiki/Deci-Platform) 🌟 NEW
+- [Comet Logging](https://github.com/ultralytics/yolov5/tree/master/utils/loggers/comet) 🌟 NEW
 
 </details>
 

diff --git a/utils/loggers/comet/__init__.py b/utils/loggers/comet/__init__.py
@@ -133,7 +133,7 @@ def __init__(self, opt, hyp, run_id=None, job_type="Training", **experiment_kwar
 
         self.comet_log_predictions = COMET_LOG_PREDICTIONS
         if self.opt.bbox_interval == -1:
-            self.comet_log_prediction_interval = self.opt.epochs // 10 if self.opt.epochs < 10 else 1
+            self.comet_log_prediction_interval = 1 if self.opt.epochs < 10 else self.opt.epochs // 10
         else:
             self.comet_log_prediction_interval = self.opt.bbox_interval