会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
爽歪歪666
不以物喜,不以己悲,努力才是永恒的主题。
博客园
首页
博问
闪存
新随笔
订阅
管理
2020年3月14日
pytorch-模型保存与加载自己训练的模型详解
摘要: filename = 'cvae_' + str(epoch+1) + '.pkl' save_path = save_dir / Path(filename) states = {} states['model'] = cvae.state_dict() # 模型参数 states['z_dim'
阅读全文
posted @ 2020-03-14 16:21 爽歪歪666
阅读(6635)
评论(0)
推荐(0)
编辑