1 Star 0 Fork 0

K-on/机器学习

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
模型的保存.py 103 Bytes
一键复制 编辑 原始数据 按行查看 历史
K-on 提交于 2020-10-26 11:05 . 模型操作
import torch
# 模型保存, 一般后缀为 .pth
torch.save(model.state_dict(), 'path/filename.pth')
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/HuangJiaLuo/machine-learning.git
[email protected]:HuangJiaLuo/machine-learning.git
HuangJiaLuo
machine-learning
机器学习
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385