PyTorch入门: Kaggle 猫狗识别VGG | Magolor

[Magolor]

https://magolor.cn/2020/01/14/2020-01-14-blog-01/#more

PyTorch入门: Kaggle 猫狗识别VGG

[zRegle]

博主你好，为啥我直接跑你的代码，loss一直都是0.69

[Magolor]

@zRegle
博主你好，为啥我直接跑你的代码，loss一直都是0.69

我一开始也遇到过这个问题，后来不记得怎么解决了。

但是我放在上面的代码都是已经解决问题的了啊……

是不是运行时间不够长？基本上要跑一段时间（以小时为单位）才会出现效果。如果还是有问题再联系我哈！

[Magolor]

@zRegle
博主你好，为啥我直接跑你的代码，loss一直都是0.69

看我当时的结果（

VGG11训练了20个epoch(使用GPU大约40分钟，不然可能要几个小时)，做到了0.34516。

VGG19训练了25个epoch(使用GPU大约三四个小时，不然可能……)，做到了0.21873。

如果GPU应该会快点……

[csfenghan]

@zRegle
博主你好，为啥我直接跑你的代码，loss一直都是0.69

我一开始也是这种情况。后来把学习率改小点多跑几个epoch后边就降下来了。

[csfenghan]

把图片读取进入内存每个epoch大概能节省一分钟多点的时间

[Magolor]

@shfenghan
把图片读取进入内存每个epoch大概能节省一分钟多点的时间

是的……我说了这个是资本家选项……内存大小不够的话没法读进来很多图片……