Cleanup

mala-project · Jul 17, 2024 · befa00d · befa00d
1 parent f9dc2be
commit befa00d
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/mala/network/trainer.py b/mala/network/trainer.py
@@ -416,8 +416,6 @@ def train_network(self):
                 t1 = time.time()
                 printout(f"training time: {t1 - t0}", min_verbosity=2)
 
-                training_loss = training_loss_sum.item() / batchid
-
                 # Calculate the validation loss. and output it.
                 torch.cuda.synchronize(
                     self.parameters._configuration["device"]
@@ -436,7 +434,6 @@ def train_network(self):
                             self.network, inputs, outputs
                         )
                         batchid += 1
-                training_loss = training_loss_sum.item() / batchid
             dataset_fractions = ["validation"]
             if self.parameters.validate_on_training_data:
                 dataset_fractions.append("train")