モデルの保存 params = net.state_dict() #netはモデル名 torch.save(params, "ファイル名.prm", pickle_protocol=4) モデルの読み込み params = torch.load("ファイル名.prm", map_location="cpu") net.load_state_dict(params)

PytorchでCNN(画像分類)を書いてみる。 以下を参考にCIFAR10 datasetを用いたCNNを書いてみる。 pytorch.org ソースコード import torch import torchvision import torchvision.transforms as transforms import torch.nn as nn import torch.nn.functional…