Salvar e carregar um modelo
Uma empresa de manufatura quer classificar seus projetos com base em imagens e decidir qual embalagem de envio é a melhor. Depois de treinar um modelo super preciso no PyTorch, você quer salvar o modelo e os pesos pré-treinados pra usar depois e compartilhar com sua equipe, garantindo que eles possam carregar tudo sem problemas.
torch e torch.nn como nn foram importados. O objeto do modelo pré-treinado está disponível na sua área de trabalho como model e sua arquitetura como ManufacturingCNN.
Este exercício faz parte do curso
Aprendizado profundo para imagens com PyTorch
Instruções do exercício
- Salve o modelo pré-treinado como um
ModelCNN.pth, lembrando de salvar os pesos, não só a arquitetura. - Crie uma instância de modelo chamada “
loaded_model” a partir da classe “ManufacturingCNN()”. - Carregue os pesos de
ModelCNN.pthemloaded_modelpassando os pesos para.load_state_dict().
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Save the model
torch.____(model.____, ____)
# Create a new model
loaded_model = ____
# Load the saved model
loaded_model.____(torch.____('ModelCNN.pth'))
print(loaded_model)