In this video I'll show you how to save and load our Neural Network Model for our Iris Neural Network with PyTorch and Python.
We need to save the weights and biases in a state_dict dictionary.
Then I'll show you how to re-load that data and use it!
#pytorch #codemy #JohnElder
Timecodes
0:00 - Introduction
1:10 - Save The NN Model
1:58 - Load the Saved Model
2:50 - Make Sure It Worked with Eval()
3:33 - Conclusion
Continue this lesson in the app
Install CourseHive on Android or iOS to keep learning while you move.
