当前位置:   article > 正文

用训练好的模型测试训练集每张图片的loss_loss小于多少结束训练

loss小于多少结束训练

用训练好的模型测试训练集每张图片的loss

所用模型为vgg 11-6.pth

computeloss.py

经过统计loss分布如下:

loss区间 个数
>10 0
1-10 1192
0.5-1 808
0.4-0.5 282
0.3-0.4 416
0.2-0.3 722
0.1-0.2 1439
0.0-0.1 31046
=0 14095
total 50000

max=9.98

min=0

将loss小于0.1的图片的索引拿出来保存在index.txt中

其中loss大于0.1的有4859张图片

将4859张图片从dataloader中拿出来不参与训练,剩下的图片训练200个epoch:

得到如下:

图片数量 训练方式 top1 error top5 error 备注
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/350279
推荐阅读
相关标签
  

闽ICP备14008679号