今日は、pytorchでニューラルネットで学習した重みを保存する方法について書きます。

前提

netが学習したネットワークモデルです。

t7ファイルで保存する方法

t7ファイルで保存する方法は、

import torch

state = {
'net': net.state_dict(), #保存するモデルの重み
'acc': acc, #精度
'epoch': epoch, #エポック数
}

torch.save(state, '保存ファイル.t7')

とまあ、こういう感じで保存します。

t7ファイルだと、重みだけでなく、精度、エポック数やいろいろ一緒に保存できるところが便利です。

いわばjsonや辞書型です。

pthファイルで保存する方法

pthファイルで保存する方法は、

import torch

torch.save(net.state_dict(), "保存ファイル.pth")

で保存します。

まとめ

後ろの.state_dict()をつけないと、gpu番号まで保存されてしまうので、違う環境で実行することがかなりむずくなったりするので、必ず.state_dict()をつけましょう。

今日は、pytorchで

前提

まず、前提条件として、t7ファイルやpthファイルで保存するときに、

エラーの解決法。

保存するときに、

「module.」が先頭について保存されている場合は、

先頭のmodule.を消してしまえばいいのです。

from collections import OrderedDict

checkpoint=torch.load("ロードしたいファイル")

state_dict=checkpoint
new_state_dict=OrderedDict()

for k, v in state_dict.items():
　　name=k[7:]; #これで先頭のmodule.を消す

　　new_state_dict[name]=v

net.load_state_dict(new_state_dict)　#これで新たにつくったorderdictを読み込み

今日は、pytorchのチュートリアルで

x = torch.empty(5, 3)
print(x)

としたときに、

error : module 'torch' has no attribute 'empty'

とエラーを吐いた時の対処法を載せます。

結論

pytorchのバージョンが古いです。

[pytorch][windows]バージョン0.3から0.4にアップデートする方法

どうもふじわらです。 今日は、pytorchのチュートリアルをやるにあたって、バージョンが0.3だと関数が「なかったので0.4にアップデートした方法を載せます。 ※以下、自己責任でお願いします pytorchのバー...

aboeblog.com

2018.10.12

こちらをみて、pytorchのバージョンをアップデートしてください。

今日は、pytorchのチュートリアルをやるにあたって、バージョンが0.3だと関数が「なかったので0.4にアップデートした方法を載せます。

※以下、自己責任でお願いします

pytorchのバージョン確認は

print(torch.__version__)

で出せます。もちろんimport torchしてからやってください（笑）

環境

Anaconda3 (64bit)

windows10

python 3.6

spyder

やり方

すでにpytorch0.3をインストールしている前提で話します。

まずは、Anaconda Promptを起動して、

conda config --add channels soumith

を入力してエンターをおしたら

conda update pytorch torchvision

を入力してエンターを押して下さい。

これでうまくいかない場合は

conda update pytorch 

でやってみてください。

もしくは、Anacondaのディレクトリのpkgsに移動して(cd ./pkgsで行けるはずです)

やってみてください。

そうしたら、version 0.4があるから、既存の0.3をアップデートするかい??って出てくると思うので、

yを入力してエンターを押して、ダウンロードとインストールされるのを待ちましょう。

以上