Zapis i wczytywanie modelu

Pewna firma produkcyjna chce klasyfikować swoje produkty na podstawie zdjęć i dobierać odpowiednie opakowania do wysyłki. Po wytrenowaniu w PyTorchu bardzo dokładnego modelu planujesz teraz zapisać model wraz z jego wstępnie wytrenowanymi wagami – zarówno do przyszłego użytku, jak i do udostępnienia zespołowi, aby wszyscy mogli go bez problemu wczytać.

torch oraz torch.nn jako nn zostały zaimportowane. Wstępnie wytrenowany obiekt modelu jest dostępny w twoim środowisku jako model, a jego architektura jako ManufacturingCNN.

To ćwiczenie jest częścią kursu

Głębokie uczenie dla obrazów z PyTorch

Zobacz kurs

Instrukcje do ćwiczenia

Zapisz wstępnie wytrenowany model jako ModelCNN.pth, pamiętając o zapisaniu wag, a nie tylko architektury.
Utwórz instancję modelu o nazwie loaded_model na podstawie klasy ManufacturingCNN().
Wczytaj wagi z pliku ModelCNN.pth do loaded_model, przekazując je do metody .load_state_dict().

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

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

# Create a new model
loaded_model = ____

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

Edytuj i uruchom kod