Keras模型训练时间太长

时间:2020-05-30 06:04:07

标签: python performance tensorflow keras lstm

因此,我具有以下用于情感分析的模型(使用预先训练的单词嵌入):
enter image description here
可见,我有一个预先训练的嵌入矩阵,并且只有约50万个可训练参数。那么,为什么要花整整一个永恒的时间来训练这种模型?批量大小为128,数量为25。第一个时间的ETA约为10分钟。我什至还没有完成。

只需提一下,我没有使用CUDA或任何东西。我认为我没有启用GPU的Tensorflow。我愿意做任何事情来提高速度。我有Tensorflow 2.1.0。

1 个答案:

答案 0 :(得分:1)

这是答案I am not using CUDA or anything。在CPU上的训练比在GPU上慢得多。如果您没有足够的高性能视频卡,则可以使用多种服务,例如Google Colab或Kaggle