Salvează și încarcă un model

O companie de producție dorește să clasifice proiectele în funcție de imagini și să determine ambalajul de expediere corespunzător. Ai antrenat un model extrem de precis în PyTorch și acum vrei să salvezi modelul împreună cu ponderile sale pre-antrenate pentru utilizare ulterioară și pentru a-l distribui echipei tale, asigurându-te că îl pot încărca fără probleme.

torch și torch.nn ca nn au fost importate. Obiectul modelului pre-antrenat este disponibil în spațiul tău de lucru ca model, iar arhitectura sa ca ManufacturingCNN.

Acest exercițiu face parte din cursul

Deep Learning pentru imagini cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

Salvează modelul pre-antrenat cu numele ModelCNN.pth, reținând că trebuie să salvezi ponderile, nu doar arhitectura.
Creează o instanță a modelului numită loaded_model din clasa ManufacturingCNN().
Încarcă ponderile din ModelCNN.pth în loaded_model transmițând ponderile către .load_state_dict().

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Save the model
torch.____(model.____, ____)

# Create a new model
loaded_model = ____

# Load the saved model
loaded_model.____(torch.____('ModelCNN.pth'))
print(loaded_model)

Editează și rulează codul