有关于如何为 .pth 文件定义模型的资源吗？

Question

错误定义为：

AttributeError: 'collections.OrderedDict' object has no attribute 'eval'

File "/text_to_speech.py", line 58, in <module>
    model = load_model(model_path)

  File "/text_to_speech.py", line 13, in load_model
    model.eval()

发生错误的代码：

def load_model(model_path):
    # Assuming model.pth is a standard PyTorch model file
    model = torch.load(model_path, map_location=torch.device('cpu'))
    model.eval()
    print("Model loaded successfully.")
    return model

我的理解是没有模型类函数定义我的 model.pth 文件。我不知道如何定义该函数以及它如何影响我的语音模型。

Answer 1

我可能误解了你的问题。但如果你有自己的类定义，这不是如何保存模型，这是正确的方法，pytorch教程。然而，在同一个教程中，还有另一种方法来保存和加载模型，在以 TorchScript 格式导出/加载模型下。