algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-24 18:45:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

code for saving the model is added

Browse Source
This commit is contained in:
yunjey
2017-03-11 14:54:46 +09:00
parent 278c13513f
commit 86a0872430
17 changed files with 630 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tutorials/01 - Linear Regression/main.py
View File

@ -58,4 +58,7 @@ predicted = model(Variable(torch.from_numpy(x_train))).data.numpy()
plt.plot(x_train, y_train, 'ro', label='Original data')
plt.plot(x_train, predicted, label='Fitted line')
plt.legend()
plt.show()
plt.show()
# Save the Model
torch.save(model, 'model.pkl')