1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

cvičení

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.

Pokyny

100 XP
  • 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().