Saving and Loading Methods
Learn how you can now save and load models as well as make predictions using class attributes.
We'll cover the following
Saving and loading
Most of the code here is the same as the code we had in the chapter, Rethinking the Training Loop. The only difference is that we will be using the class attributes instead of the local variables.
The updated method for saving checkpoints should look like this now:
Get hands-on with 1200+ tech skills courses.