Збереження та завантаження моделі

Виробнича компанія хоче класифікувати свої проєкти за зображеннями та визначати відповідну пакувальну тару для відправлення. Ви натренували в PyTorch дуже точну модель і тепер плануєте зберегти модель і її попередньо навчені ваги для подальшого використання та щоб поділитися з командою, забезпечивши можливість безперешкодно завантажувати її.

torch і torch.nn як nn уже імпортовано. Попередньо навчена модель доступна у вашому середовищі як model, а її архітектура — як ManufacturingCNN.

Ця вправа є частиною курсу

Глибоке навчання для зображень із PyTorch

Переглянути курс

Інструкції до вправи

Збережіть попередньо навчену модель як ModelCNN.pth, не забудьте зберегти саме ваги, а не лише архітектуру.
Створіть екземпляр моделі з назвою loaded_model на основі класу ManufacturingCNN().
Завантажте ваги ModelCNN.pth у loaded_model, передавши їх до .load_state_dict().

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

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

# Create a new model
loaded_model = ____

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

Редагувати та запускати код