pytorchの事前学習済みモデルの使い方

実現したいこと

発生している問題・エラーメッセージ

resnet18を動作させようとすると、次のエラーが発生しました。

FileNotFoundError: [Errno 2] No such file or directory: '/mnt/lustre/weixiuying/model_zoo/resnet18_imagenet.pth.tar'

学習済みモデルを指定のパスに保存しなければいけない、と解釈しています。

resnet.pyでモデルを構築し、hubconf.pyでモデルをロードしているようです。hubconf.pyの12行目でmodel_pathを

'resnet18': prefix+'/model_zoo/resnet18_imagenet.pth.tar',

と指定していますが、この場所にモデルを保存するためにはどうすればよいのでしょうか。

補足情報(FW/ツールのバージョンなど)

Python 3.9.16
torch 2.0.1

コメントを投稿

0 コメント