1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie dla obrazów z PyTorch
Connected

ćwiczenie

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.

Instrukcje

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