训练一个epoch 一个小时都出不来 我看了cpu和gpu的占用都相当低10%以内 但是代码请人看过了没啥毛病 现在这个问题可以怎么解决呢?
不用vgg16就好使 不造为啥
试试看把batch size调整大一些看看。再不行减少下数据量
可以加一些输出语句看看看卡在哪了,是数据加载和预处理还是训练
可以考虑使用多线程或批处理方式来加快数据加载
考虑使用更简单的网络结构,或者减少网络的层数、节点数等来加快训练速度。
如果你使用的是小批量(mini-batch)训练,而且批处理大小设置过大,可能会导致训练速度变慢。尝试减小批处理大小,以加快训练进程。
数据读取和模型或者代码优化等方面考虑一下
参考
(1) Pytorch 加速数据读取 - 知乎 https://zhuanlan.zhihu.com/p/80695364
(2) 模型训练速度过慢,GPU利用率低_aistudio训练速度太慢 https://blog.csdn.net/swocky/article/details/105922049
(3) 关于PyTorch训练深度学习模型CPU/GPU使用率低,训练速度慢这一问题的总结_cpu模型训练速度很慢 https://blog.csdn.net/weixin_40227656/article/details/119741324
运行一下我这个试试http://t.csdn.cn/sV7oX