【PyTorch】AttributeError: 'GeneralizedRCNNTransform' object has no attribute 'fixed_size' - Qiita

TL;DR GPU環境で学習したモデルをtorch.saveで保存し、CPU環境でtorch.loadで読み出し推定を行うとエラーが発生する場合がる。 解決方法 torch.loadではなく各種学習モデルのクラスからインスタンス化してload_state_dictで学習したパラメータを読み出す。 詳細 [PyTorch] TORCHVISION OBJECT DETECTION FINETUNING TUTORIALを基に物体検出を試す 環境 CPU側 anaconda3 Python 3.8.10 pytorch 1.10.0-py3.8_cpu_0 torchvision 0.11.1-py38_cpu 原因 torch.save, torch.loadでモデルを保存・読み込みすると、__init__()経由でインスタンス化されずNoneのクラスメンバが未定義となってしまう。結果として