- 博客(4)
- 收藏
- 关注
原创 ubantu16.04+nvidia440.100+cuda10.2+cudnn7-7.6.5+torch1.5.0
※1 小型服务器配置:GPU:2块型号:GeForec 1080Ti※2 驱动配置: NVIDIA: 440.100CUDA:10.2Cudnn:cudnn7-7.6.5Torch:pip install torch==1.5.0+cu102 torchvision==0.6.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html※3 驱动下载NVIDIA-Linux-x86_64-440.1...
2021-09-20 15:42:25 226
原创 错误现象:pytorch训练分类网络,所有样本分类结果一样
具体现象1:训练alexnet分类网络,多次epoch后,分类精度为同样的值,而且精度很低原因: 发现所有样本都预测为同一类别进一步检查,具体现象2:在第一个epoch的第一个step中,pred_y(预测分类)的结果正常(概率形式),但第二个step以及以后的step,pred_y的结果不正常([0,0,1,0,....],即某1位为1,其它位全部为0)原因:第一个step时,网络还未优化;但第二个step时网络已优化,所以考虑优化错误。解决策:改小学习率(lr:0.001 .
2021-08-09 15:04:01 2582
原创 错误现象:pytorch训练分类网络,acc为0
原因:计算pred_Y时,return多缩进了一个tab,导致只输出一个batchsize的pred_Y前无古人后无来者的导致错误方式...
2021-08-07 18:26:04 863
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人