multiply loss by 100 when exporting log to improve readability

choderalab · Mar 14, 2024 · d6fea04 · d6fea04
1 parent 790df53
commit d6fea04
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/espfit/app/train.py b/espfit/app/train.py
@@ -345,7 +345,7 @@ def report_loss(self, epoch, loss_dict):
 
         log_file_path = os.path.join(self.output_directory_path, 'reporter.log')
         df_new = pd.DataFrame.from_dict(loss_dict, orient='index').T
-        df_new.mul(100)   # Multiple each loss component by 100
+        df_new = df_new.mul(100)   # Multiple each loss component by 100
         df_new.insert(0, 'epoch', epoch)
 
         if os.path.exists(log_file_path):