Merge pull request #114 from xinqiu/patch-1

fix spelling mistake
This commit is contained in:
Yunjey Choi
2018-05-15 22:47:21 +09:00
committed by GitHub

View File

@ -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')