Uložení a načtení modelu

Výrobní společnost chce klasifikovat své projekty na základě obrázků a určovat vhodné balení pro přepravu. Máš za sebou trénink vysoce přesného modelu v PyTorchi a teď ho chceš uložit spolu s natrénovanými váhami pro budoucí použití a sdílet ho s týmem – tak, aby ho všichni mohli bez problémů načíst.

torch a torch.nn jako nn jsou již naimportovány. Natrénovaný objekt modelu je dostupný v pracovním prostředí jako model a jeho architektura jako ManufacturingCNN.

Toto cvičení je součástí kurzu

Deep Learning pro obrázky s PyTorchem

Zobrazit kurz

Pokyny k cvičení

Ulož natrénovaný model jako ModelCNN.pth – nezapomeň uložit váhy, nejen architekturu.
Vytvoř instanci modelu s názvem loaded_model z třídy ManufacturingCNN().
Načti váhy ze souboru ModelCNN.pth do loaded_model předáním vah metodě .load_state_dict().

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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

# Create a new model
loaded_model = ____

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

Upravit a spustit kód