Spara och läs in en modell
Ett tillverkningsföretag vill klassificera sina produkter utifrån bilder och avgöra lämplig förpackning för frakt. Du har tränat upp en modell med hög precision i PyTorch och vill nu spara modellen och dess förtränade vikter för framtida bruk och för att dela med ditt team, så att de enkelt kan läsa in den.
torch och torch.nn som nn har importerats. Det förtränade modellobjektet finns tillgängligt i din workspace som model, och dess arkitektur som ManufacturingCNN.
Den här övningen är en del av kursen
Djupinlärning för bilder med PyTorch
Övningsinstruktioner
- Spara den förtränade modellen som
ModelCNN.pthoch kom ihåg att spara vikterna, inte bara arkitekturen. - Skapa en modellinstans med namnet
loaded_modelfrån klassenManufacturingCNN(). - Läs in vikterna från
ModelCNN.pthtillloaded_modelgenom att skicka vikterna till.load_state_dict().
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Save the model
torch.____(model.____, ____)
# Create a new model
loaded_model = ____
# Load the saved model
loaded_model.____(torch.____('ModelCNN.pth'))
print(loaded_model)