Fix bug

cleardusk · May 4, 2019 · 5e2d051 · 5e2d051
1 parent dea30ce
commit 5e2d051
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -168,7 +168,7 @@ def validate(val_loader, model, criterion, epoch):
             output = model(input)
 
             loss = criterion(output, target)
-            losses.append(loss)
+            losses.append(loss.item())
 
         elapse = time.time() - end
         loss = np.mean(losses)