- 博客(2)
- 资源 (1)
- 收藏
- 关注
RSS订阅原创 Pytorch：定义的网络结构层能否重复使用
**前言：**最近在构建网络的时候，有一些层参数一样，于是就没有定义新的层，直接重复使用了原来已经有的层，发现效果和模型大小都没有什么变化，心中产生了疑问：定义的网络结构层能否重复使用？因此接下来利用了一个小模型网络实验了一下。...
2019-01-23 17:05:25
6124
6
原创 Pytorch：模型的保存与加载
模型保存与加载常用有两种方法，第一种是保存整个模型，包括模型的结构和参数；第二种是保存模型的参数。推荐使用第二种，因为模型一旦很大，第一种加载耗时长，其次第二种加载方式更加灵活，可以加载其他模型的预训练参数，从而使用迁移学习的方法减小训练时长。一、保存/加载整个模型保存模型：torch.save(net, ‘model_net1.pkl’)加载模型net_parm = ‘model_...
2019-01-21 13:55:37
461
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人