fix(tune): undef train_loss

jyaacoub · Dec 6, 2023 · f8f80f6 · f8f80f6
1 parent e3205cb
commit f8f80f6
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/train_test/tune.py b/src/train_test/tune.py
@@ -47,7 +47,6 @@ def simple_train(model: BaseModel, optimizer:torch.optim.Optimizer,
 
             # Compute loss
             loss = CRITERION(predictions, labels)
-            train_loss += loss.item()
 
             # Backward pass and optimization
             optimizer.zero_grad()