y-kamiya https://blog.hatena.ne.jp/y-kamiya/ MEMOcho- https://jsapachehtml.hatenablog.com/ gpuで学習して出力したデータを、cpuでloadしようとしたらタイトルのようなエラーが出た フォーラムで同じ問題が質問されていたので簡単に解決 Unexpected key in state_dict: "bn1.num_batches_tracked" - PyTorch Forums こんな感じでload_state_dictにstrict=Falseを指定するだけ load_state_dict(torch.load(path_to_data, map_location=device_name), strict=False) 余分なキーが含まれていても無視してくれるオプションのよう t… 190 <iframe src="https://hatenablog-parts.com/embed?url=https%3A%2F%2Fjsapachehtml.hatenablog.com%2Fentry%2F2019%2F03%2F02%2F185035" title="pytorchでエラー(Unexpected key(s) in state_dict: &quot;bn1.num_batches_tracked&quot;) - MEMOcho-" class="embed-card embed-blogcard" scrolling="no" frameborder="0" style="display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;"></iframe> Hatena Blog https://hatena.blog 2019-03-02 18:50:35 rich https://jsapachehtml.hatenablog.com/entry/2019/03/02/185035 1.0 100%