Keras模型训练时间过长

biswetbf  于 2024-01-08  发布在  其他
关注(0)|答案(1)|浏览(132)

因此,我有以下情感分析模型(使用预训练的词嵌入):


的数据
可以看到,我有一个预训练的嵌入矩阵,只有大约50万个可训练参数。那么为什么训练这个模型需要一个漫长的过程呢?批量大小是128,epoch的数量是25。第一个epoch的ETA大约是10分钟。我甚至还没有完成。
顺便提一下,我没有使用CUDA或其他任何东西。我不认为我有一个支持GPU的Tensorflow。我愿意做任何事情来提高速度。我有Tensorflow 2.1.0。

b09cbbtk

b09cbbtk1#

这里是答案I am not using CUDA or anything。CPU上的训练比GPU上的慢得多。如果你没有足够高性能的显卡,你可以使用谷歌Colab或Kaggle等几个服务

相关问题