Rename train_loss -> total_train_loss

Signed-off-by: Kim, Vinnam <vinnam.kim@intel.com>
openvinotoolkit · Nov 28, 2023 · 6128725 · 6128725
1 parent 61c09ab
commit 6128725
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/otx/core/model/module/base.py b/src/otx/core/model/module/base.py
@@ -58,15 +58,15 @@ def training_step(self, inputs: OTXBatchDataEntity, batch_idx: int) -> Tensor:
                     prog_bar=True,
                 )
 
-            train_loss = sum(train_loss.values())
+            total_train_loss = sum(train_loss.values())
             self.log(
                 "train/loss",
-                train_loss,
+                total_train_loss,
                 on_step=True,
                 on_epoch=False,
                 prog_bar=True,
             )
-            return train_loss
+            return total_train_loss
 
         raise TypeError(train_loss)