なんとな~くしあわせ?の日記

「そしてそれゆえ、知識そのものが力である」 (Nam et ipsa scientia potestas est.) 〜 フランシス・ベーコン

2020-08-18から1日間の記事一覧

pytorchでGPUで訓練したモデルをCPUで動かす時

github.com表題通り pytorchは機械学習モデルの学習にCPUを使うかGPUを使うか選べる pythonでは機械学習モデルをpickle形式で保存することがある pytorchでGPUで訓練したモデルをCPUで動かす時、起こるエラーとその対処 以下のようにtorch.loadに引数を与え…