mirror of
https://github.com/yunjey/pytorch-tutorial.git
synced 2025-07-04 16:12:12 +08:00
fix spelling mistake
This commit is contained in:
@ -59,7 +59,7 @@ for epoch in range(num_epochs):
|
||||
.format(epoch+1, num_epochs, i+1, total_step, loss.item()))
|
||||
|
||||
# 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():
|
||||
correct = 0
|
||||
total = 0
|
||||
@ -73,4 +73,4 @@ with torch.no_grad():
|
||||
print('Accuracy of the model on the 10000 test images: {} %'.format(100 * correct / total))
|
||||
|
||||
# Save the model checkpoint
|
||||
torch.save(model.state_dict(), 'model.ckpt')
|
||||
torch.save(model.state_dict(), 'model.ckpt')
|
||||
|
Reference in New Issue
Block a user