加载模型AttributeError: 'dict' object has no attribute 'eval'

### 错误分析与解决方案 在加载模型时出现的 `AttributeError: 'dict' object has no attribute 'eval'` 错误通常是因为模型保存时仅保存了模型的状态字典（`state_dict`），而没有保存完整的模型结构。当尝试调用 `.eval()` 方法时，由于状态字典是一个普通的 Python 字典对象，并不包含模型的结构信息，因此会抛出上述错误。 以下是详细的解决方案： #### 1. 确保正确保存模型 如果仅需要保存模型的权重，可以使用以下代码保存模型的状态字典： ```python torch.save(model.state_dict(), model_path) ``` 但需要注意的是，这种方式仅保存了模型的权重，而不包含模型的结构。如果需要保存完整的模型结构和权重，可以使用以下代码： ```python torch.save(model, model_path) ``` #### 2. 加载模型时的正确方法 - **如果保存的是状态字典**：需要先实例化模型，然后加载权重。 ```python model = YourModelClass() # 实例化模型 model.load_state_dict(torch.load(model_path)) # 加载权重 model.eval() # 调用 eval 方法切换到评估模式 ``` 注意：这里不能直接将 `load_state_dict` 的返回值赋给变量[^3]，因为该函数返回的是一个 `_IncompatibleKeys` 对象，而不是模型本身[^4]。 - **如果保存的是完整模型**：可以直接加载模型并调用 `.eval()` 方法。 ```python model = torch.load(model_path) # 加载完整模型 model.eval() # 调用 eval 方法切换到评估模式 ``` #### 3. 检查加载的模型类型 如果不确定保存的文件是状态字典还是完整模型，可以通过以下代码检查： ```python import torch data = torch.load(model_path) if isinstance(data, dict): # 如果是状态字典 print("保存的是状态字典") else: print("保存的是完整模型") ``` #### 4. 常见问题排查 - 如果仍然出现类似 `AttributeError: '_IncompatibleKeys' object has no attribute 'eval'` 的错误，可能是加载权重后未正确调用 `.eval()` 方法[^4]。确保加载权重后立即调用 `.eval()` 方法。 - 如果保存的文件中包含额外的键值对（例如 `'state_dict'`），需要提取正确的键值。例如： ```python checkpoint = torch.load(model_path) model.load_state_dict(checkpoint['state_dict']) model.eval() ``` --- ### 示例代码 以下是一个完整的加载模型示例，假设保存的是状态字典： ```python import torch from your_model_module import YourModelClass # 替换为你的模型定义 # 加载模型 model = YourModelClass() # 实例化模型 model.load_state_dict(torch.load(model_path)) # 加载权重 model.eval() # 切换到评估模式 ``` 如果保存的是完整模型： ```python import torch # 加载模型 model = torch.load(model_path) # 直接加载完整模型 model.eval() # 切换到评估模式 ``` ---