pytorch（三）——保存自己的模型

Jack_and_monkey

于 2019-02-14 11:17:28 发布

阅读量884

点赞数

CC 4.0 BY-SA版权

分类专栏： PyTorch Python

本文链接：https://blog.youkuaiyun.com/Jack_and_monkey/article/details/87259330

7 篇文章

订阅专栏

6 篇文章

订阅专栏

本文详细介绍如何在PyTorch中保存模型参数及整个模型，包括两种保存方法的具体操作步骤，以及可能出现的错误及其解决办法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


torch.save(the_model.state_dict(), PATH)


the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))


torch.save(the_model, PATH)

the_model = torch.load(PATH)

AttributeError: Can't get attribute 'Net' on module '__main__'

model = the_model()
print(model)

如果能打印出網絡結構，則成功了～～