AttributeError：“ collections.OrderedDict”对象没有属性“ eval” [英] AttributeError: 'collections.OrderedDict' object has no attribute 'eval'

问题描述

我有一个看起来像这样的模型文件

I have a model file which looks like this

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（）时，我会得到，

I want to do inference on this model, but when i do model.eval() i get,

AttributeError：'collections.OrderedDict'对象没有属性'eval
不确定如何处理对此，关于如何解决此问题的任何建议将非常有帮助，感谢advan ce

AttributeError: 'collections.OrderedDict' object has no attribute 'eval Not really sure how to go about this, any suggestions on how i can get this fixed will be really helpful,Thanks in advance

推荐答案

它不是模型文件，而是状态文件。在模型文件中，存储了完整的模型，而在状态文件中，仅存储了参数。

因此，您的 OrderedDict 只是用于您的模型。您将需要创建模型，然后将这些值加载到模型中。因此，该过程将采用以下形式：

It is not a model file, instead, this is a state file. In a model file, the complete model is stored, whereas in a state file only the parameters are stored.
So, your OrderedDict are just values for your model. You will need to create the model and then need to load these values into your model. So, the process will be something in form of

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()

您需要正确定义模型。上例中给出的只是一个虚拟对象。如果您自己构建模型，则可能需要更新保存的dict文件的密钥，如此处。最好的做法是使用与保存 state_dict 时完全相同的方式定义模型，然后直接执行 model.load_state_dict 将起作用。

You'll need to define your model properly. The one given in the example above is just a dummy. If you construct your model yourself, you might need to update the keys of the saved dict file as mentioned here. The best course of action is to define your model in exactly the same way from when the state_dict was saved and then directly executing model.load_state_dict will work.

这篇关于AttributeError：“ collections.OrderedDict”对象没有属性“ eval”的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！