Fixed variable name in docstr to add_yaml_summary

Deci-AI · Oct 23, 2023 · 6b3999c · 6b3999c
1 parent 58c96ca
commit 6b3999c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/super_gradients/common/sg_loggers/base_sg_logger.py b/src/super_gradients/common/sg_loggers/base_sg_logger.py
@@ -337,7 +337,7 @@ def add_yaml_summary(self, tag: str, summary_dict: dict, global_step: Optional[i
         but who knows what it will be suited for later.
 
         :param tag:         Identifier of the summary.
-        :param state_dict:  Checkpoint state_dict.
+        :param summary_dict:  Checkpoint summary_dict.
         :param global_step: Epoch number.
         """