训练模型出现了一个问题

img


训练了79个 ,每当运行到79个,80个就会出现问题(我是新入门)

该回答引用GPTᴼᴾᴱᴺᴬᴵ
这个错误提示显示了一个属性错误,即 'AutoDriveNet' 对象没有 'module' 属性。从代码片段来看,这可能是使用了 PyTorch 的 nn.DataParallel 模块将模型并行化的结果,因为当使用该模块时,模型对象被包装在另一个对象中,而不是直接使用模型对象。因此,访问模型权重时需要使用 .module 属性。

为了解决这个问题,你可以在访问模型权重时使用 .module 属性,例如:

model.module.state_dict()


如果这个解决方法不行,你可以考虑检查你的代码,特别是训练过程中的配置,以查找其他可能导致这个错误的问题。