pytorch
关注
分享
扫一扫
文章平均质量分 80
Decheng.G
这个作者很懒，什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
backbone.bn1.num_batches_tracked 报错
加载model一直报错unexpected backbone.bn1.num_batches_tracked原来是使用的版本的问题，pytorch0.4.1之后在BN层加入了track_running_stats这个参数这个参数的作用如下:训练时用来统计训练时的forward过的min-batch数目,每经过一个min-batch, track_running_stats+=1如果没有指定...原创 2019-11-15 09:36:15 · 1437 阅读 · 0 评论 -
Pytorch(cuda9.0 cudnn7.05 python2.7 conda)离线安装
Pytorch+cuda9.0+cudnn7.05+python2.7+conda离线安装官网conda指令安装 conda install pytorch torchvision cuda90 -c pytorch 但是conda网站并不稳定，有时候可能会造成连接错误，所以尝试了离线安装。 Pytorch（cuda9.0）主要依赖以下几个包 cuda90-1.0-h6433d27_0...原创 2018-09-15 19:16:00 · 6511 阅读 · 0 评论