【发布时间】:2021-01-10 18:38:55
【问题描述】:
我正在测试我的 DenseNet-121 实现是否可以在 Ubuntu 上的 GPU (RTX 2060) 上运行,但是当它成功运行时,我观察到损失和准确度的表现明显低于使用不同计算机在 CPU 上设置的性能。
我使用相同的代码进行模型和数据准备、相同的数据集和相同的训练参数。
唯一的区别是令人满意的 macOS CPU 版本运行在 TensorFlow 版本 2.1.0 上,而 Ubuntu GPU 版本运行在 TensorFlow 版本 2.4.0 上。
为了找出唯一可观察到的差异,我使用了 macOS 版本并将 TensorFlow 更新到 2.4.0 版本,现在我记录的结果与在其他设备上一样明显更差。
1 epoch 后的 TensorFlow 版本 2.1.0:
593/593 [==============================] - 733s 1s/sample - loss: 0.3325 - accuracy: 0.8650
1 epoch 后的 TensorFlow 版本 2.4.0:
38/38 [==============================] - 823s 21s/step - loss: 0.3732 - accuracy: 0.3920
*这里最明显的是新 2.4.0 和 2.1.0 之间相同过程的 0.473 精度差异。
有人知道为什么会出现这个问题吗?最简单的解决方案是回到上一个 TensorFlow 版本并获得满意的结果吗?
提前谢谢你!
【问题讨论】:
标签: python performance tensorflow keras version