Save and Load our Neural Network Model - Deep Learning with PyTorch 9

Показать описание

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

Рекомендации по теме

Комментарии

▶ Watch Deep Learning With Pytorch Playlist ✅ Subscribe To My YouTube Channel:
▶ See More At: ✅ Join My Facebook Group:
▶ Get The Code

Codemycom

I really love the simplicity with which you teach these. I had done my theories long back but it was so difficult to get some coding sessions and found yours to the save.

datacodedesign

finally got a neural network working, thanks

alexmarcellinus

So we wont be able the loss value same as the one we got while training 100 epochs?

Ragul_SL

The in_features and out_features are exactly the same as what we started with shouldn't they be different?

NormalCasinoWins

If anyone getting the " RuntimeError: expected scalar type Long but found Float", then change y labels to "LongTensor" instead of FloatTensor.

y_train = torch.LongTensor(y_train)
y_test = torch.LongTensor(y_test)

JayA-qslc

Is there a reason you save just the state of a model and not the entire model itself?
I was specifically thinking you could do:
torch.save(model, "Iris Model")
I felt like this could be more useful if you were bringing the model to another program but I was curious if there were any disadvantages to doing this?

Tacocat

I tried using pytorch to train a logistic regression model and the experience was 🥲🥲

pipi_delina

Save and Load our Neural Network Model - Deep Learning with PyTorch 9

Save and Load our Neural Network Model - Deep Learning with PyTorch 9

Saving and Loading a PyTorch Neural Network (3.3)

[05] Making Neural Network in GMS2 - Buffer Save & Load

[04] Making Neural Network in GMS2 - JSON Save & Load

Neural DSP Presets | How to Save, Send, and Load Your Tones

Saving and Loading Models (Coding TensorFlow)

Neural networks tutorial: Fully Connected 10 [Java] - Saving and loading

Loading in your own data - Deep Learning basics with Python, TensorFlow and Keras p.2

How to Build a Neural Network with TensorFlow and Keras in 10 Minutes

How to Save, Load, and Test Models in Keras and TensorFlow (Neural Networks for DH 11d)

[6] Ways to save the neural network in PyTorch

Load Data and Train Neural Network Model - Deep Learning with PyTorch 6

Save Neural Network in JSON file | Save Neural Network model | Technique 2 | Machine Learning

Load your trained convolutional neural network saved as an h5 file to predict 7 types of emotion

Train Neural Network by loading your images |TensorFlow, CNN, Keras tutorial

Building a Python Neural Network Image Classification Part 4 :Train,Test,Save & Load Model

Neural Network Tutorial - Ch. 8.1 Save method (Part 1)

Neural Network Tutorial - Ch. 8.2 Save method (Part 2)

You can grow new brain cells. Here's how | Sandrine Thuret | TED

ml5: Save Neural Network Trained Model

Neural Networks From Zero: Custom Data

Neural Network Tutorial - Ch. 9.1 Load method (Part 1)

How Neural Networks Can Be Hacked (And What You Should Do To Protect It)!

Part 05 - Constructing a Neural Network Models - Regression model with Python (Tensorflow & Kera...