web-dev-qa-db-ja.com

AttributeError： 'collections.OrderedDict'オブジェクトには属性 'eval'がありません

このようなモデルファイルがあります

OrderedDict([('inp.conv1.conv.weight', 
          (0 ,0 ,0 ,.,.) = 
           -1.5073e-01  6.4760e-02  1.9156e-01
            1.2175e-01  3.5886e-02  1.3992e-01
           -1.5903e-01  8.2055e-02  1.7820e-01

          (0 ,0 ,1 ,.,.) = 
            1.0604e-01 -1.3653e-01  1.4803e-01
            6.0276e-02 -1.4674e-02  2.3059e-06
           -6.2192e-02 -5.1061e-03 -7.4145e-03

          (0 ,0 ,2 ,.,.) = 
           -5.5632e-02  3.5326e-02  6.5108e-02
            1.1411e-01 -4.4160e-02  8.2610e-02
            8.9979e-02 -3.5454e-02  4.2549e-02

          (1 ,0 ,0 ,.,.) = 
            4.8523e-02 -4.3961e-02  5.3614e-02
           -1.2644e-01  1.2777e-01  8.9547e-02
            3.8392e-02  2.7016e-02 -1.4552e-01

          (1 ,0 ,1 ,.,.) = 
            9.5537e-02  2.8748e-02  3.9772e-02
           -6.2410e-02  1.1264e-01  7.8663e-02
           -2.6374e-02  1.4401e-01 -1.7109e-01

          (1 ,0 ,2 ,.,.) = 
            5.1791e-02 -1.6388e-01 -1.7605e-01
            3.5028e-02  7.7164e-02 -1.4499e-01
           -2.9189e-02  2.7064e-03 -2.3228e-02

          (2 ,0 ,0 ,.,.) = 
           -7.4446e-03 -9.7202e-02 -1.4704e-01
           -1.0019e-02  8.1780e-02 -5.3530e-02
           -1.8412e-01  1.5988e-01 -1.3450e-01

          (2 ,0 ,1 ,.,.) = 
           -1.1075e-01 -5.2478e-02  6.0658e-02
            1.6739e-01 -2.9360e-02  1.2621e-01
            2.0686e-02  1.1468e-01  1.2282e-01

私はこのモデルで推論をしたいのですが、model.eval（）を実行すると、

AttributeError: 'collections.OrderedDict' object has no attribute 'evalこれをどうやって進めるかわからない、この問題を解決する方法についての提案は本当に役立つでしょう、事前に感謝します

deep-learningpytorch
6
Ryan

これはモデルファイルではなく、状態ファイルです。モデルファイルには完全なモデルが保存されますが、状態ファイルにはパラメーターのみが保存されます。
したがって、OrderedDictはモデルの単なる値です。モデルを作成してから、これらの値をモデルにロードする必要があります。そのため、プロセスは次のような形になります

import torch
import torch.nn as nn

class TempModel(nn.Module):
    def __init__(self):
        self.conv1 = nn.Conv2d(3, 5, (3, 3))
    def forward(self, inp):
        return self.conv1(inp)

model = TempModel()
model.load_state_dict(torch.load(file_path))
model.eval()

モデルを適切に定義する必要があります。上記の例で示したものは単なるダミーです。自分でモデルを構築する場合、 here のように保存されたdictファイルのキーを更新する必要があるかもしれません。アクションの最善の方法は、state_dictが保存されたときとまったく同じ方法でモデルを定義し、model.load_state_dictを直接実行すると機能することです。

11
layog