torch.save 时报错 Can‘t pickle local object ‘

    科技2025-07-15  15

    保存模型

    虽然没有明白是什么原因,但是找到了解决方法:

    将torch.save(model, path) # 直接保存整个模型方法改为torch.save(model.state_dict(), path) # 保存模型的参数

    载入模型

    相应的,载入模型时将model = torch.load(path) # 直接加载模型方法改为 model = Model() # 先初始化一个模型 model.load_state_dict(torch.load(path)) # 再加载模型参数

    Processed: 0.011, SQL: 8