Pytorchのnn.DataParallelを使ったモデルを保存するとloadするときにエラーになる問題 - QiitaPytorchはMultiGPUで学習・推論するときの便利な機能として torch.nn.DataParallelというモジュールを準備してくれている。 すごく楽に使えるのですが、一方でこれで学習したモデルを保存・利用する際にちょっと気をつけないと行けない点があるので忘れないうちにメモしておく。 なぜ起こるのか？ ここで、 self.module = module が入っているので、Model構成のprefixが元のModel Classと変わるため。 ミスると下記のようなエラーが出る。
1
リリース、障害情報などのサービスのお知らせ
最新の人気エントリーの配信
処理を実行中です
j次のブックマーク
k前のブックマーク
lあとで読む
eコメント一覧を開く
oページを開く
