2019-03-02から1日間の記事一覧

pytorchでエラー(Unexpected key(s) in state_dict: "bn1.num_batches_tracked")

gpuで学習して出力したデータを、cpuでloadしようとしたらタイトルのようなエラーが出た フォーラムで同じ問題が質問されていたので簡単に解決 Unexpected key in state_dict: "bn1.num_batches_tracked" - PyTorch Forums こんな感じでload_state_dictにstr…