モデルの保存と読み込み

製造業の会社が、画像に基づいてプロジェクトを分類し、適切な出荷用の梱包を決めようとしています。PyTorch で高精度なモデルを学習できたので、将来の再利用やチームへの共有に向けて、モデル本体と事前学習済みの重みを保存し、だれでもシームレスに読み込めるようにしたいと考えています。

torch と torch.nn（nn）はインポート済みです。事前学習済みのモデルオブジェクトは model、そのアーキテクチャは ManufacturingCNN としてワークスペースに用意されています。