如何在GPU上更快地进行深度学习模型训练?

时间:2019-10-25 09:56:58

标签: python tensorflow deep-learning tensorflow2.0

我曾尝试使用Tensorflow 2.0训练VGG16模型以在Titan RTX GPU(24G内存)上获取图像网络数据。

但是,预计培训时间约为52天。(批量为128个,共300个纪元)

这是正常情况吗?我觉得这很奇怪。

而且,我能知道有什么方法可以加速训练我的模型吗?

gist是我的全部代码。谢谢。

0 个答案:

没有答案