From 6b3999cf407b03c98f818d8ff89c5f149d144744 Mon Sep 17 00:00:00 2001
From: hakuryuu96 <marchenkophilip@gmail.com>
Date: Mon, 23 Oct 2023 10:57:48 +0000
Subject: [PATCH] Fixed variable name in docstr to add_yaml_summary

---
 src/super_gradients/common/sg_loggers/base_sg_logger.py | 2 +-
 1 file changed, 1 insertion(+), 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
index 379b34169f..bd2156ca82 100644
--- 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.
         """