mirror of
https://github.com/yunjey/pytorch-tutorial.git
synced 2025-07-07 10:05:29 +08:00
@ -59,7 +59,7 @@ for epoch in range(num_epochs):
|
|||||||
.format(epoch+1, num_epochs, i+1, total_step, loss.item()))
|
.format(epoch+1, num_epochs, i+1, total_step, loss.item()))
|
||||||
|
|
||||||
# Test the model
|
# Test the model
|
||||||
# In test phase, we don't need to compute gradients (for memory efficieny)
|
# In test phase, we don't need to compute gradients (for memory efficiency)
|
||||||
with torch.no_grad():
|
with torch.no_grad():
|
||||||
correct = 0
|
correct = 0
|
||||||
total = 0
|
total = 0
|
||||||
|
Reference in New Issue
Block a user