会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
华小电
新随笔
管理
2024年7月19日
Pytorch模型文件`*.pt`与`*.pth` 的保存与加载
摘要: 1.*.pt文件 .pt文件保存的是模型的全部,在加载时可以直接赋值给新变量model = torch.load("filename.pt")。 具体操作: (1). 模型的保存 torch.save(model,"Path/filename.pt") (2). 模型的加载 model = torc
阅读全文
posted @ 2024-07-19 10:09 华小电
阅读(354)
评论(0)
推荐(0)
编辑