【问题标题】:Keras model taking too long to trainKeras 模型训练时间过长
【发布时间】:2020-05-30 06:04:07
【问题描述】:

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

可见,我有一个预训练的嵌入矩阵,只有大约 50 万个可训练参数。那么为什么训练这个模型需要很长时间呢?批量大小为 128,epoch 数为 25。第一个 epoch 的 ETA 约为 10 分钟。我什至还没有完成。

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

【问题讨论】:

  • 没什么可说的,你没有包括任何重要的细节。 “一整个永恒”是多少时间?你用哪个 CPU 来训练这个?

标签: python performance tensorflow keras lstm


【解决方案1】:

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

【讨论】:

  • 我有 RTX 2070。会有帮助吗?
  • 嗯,这绝对是一张高性能的卡。要在 tensorflow 中启用 GPU 支持,请使用 tensorflow.org/install/gpu
猜你喜欢
  • 2021-01-12
  • 1970-01-01
  • 2019-01-09
  • 2021-08-07
  • 2017-07-28
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
相关资源
最近更新 更多