fix throttling

jynx10 · Aug 10, 2021 · 80b7503 · DN6 · Aug 10, 2021 · DN6
1 parent e4ad51f
commit 80b7503
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/catalyst/loggers/comet.py b/catalyst/loggers/comet.py
@@ -165,12 +165,13 @@ def log_metrics(
         prefix_parameters = [stage_key, loader_key, scope]
         prefix = self._format_prefix(prefix_parameters)
 
-        self.experiment.log_metrics(
-            metrics,
-            step=global_batch_step,
-            epoch=global_batch_step,
-            prefix=prefix,
-        )
+        if global_batch_step % 10 == 0:
+            self.experiment.log_metrics(
+                metrics,
+                step=global_batch_step,
+                epoch=global_batch_step,
+                prefix=prefix,
+            )
 
     def log_image(
         self,