Keras 模型训练时间太长
Keras model taking too long to train
所以我有以下情绪分析模型(使用预训练词嵌入):
如图所示,我有一个预训练的嵌入矩阵和大约 500k 的可训练参数。那么为什么训练这个模型需要整个永恒呢?批量大小为 128,轮数为 25。第一个轮数的预计到达时间约为 10 分钟。我什至还没有完成。
顺便提一下,我没有使用 CUDA 或任何东西。我认为我没有启用 GPU 的 Tensorflow。我愿意做任何事情来提高速度。我有 Tensorflow 2.1.0.
这是答案 I am not using CUDA or anything
。在 CPU 上的训练比在 GPU 上慢得多。如果你没有足够高性能的显卡,你可以使用一些服务,例如 Google Colab 或 Kaggle
所以我有以下情绪分析模型(使用预训练词嵌入):
如图所示,我有一个预训练的嵌入矩阵和大约 500k 的可训练参数。那么为什么训练这个模型需要整个永恒呢?批量大小为 128,轮数为 25。第一个轮数的预计到达时间约为 10 分钟。我什至还没有完成。
顺便提一下,我没有使用 CUDA 或任何东西。我认为我没有启用 GPU 的 Tensorflow。我愿意做任何事情来提高速度。我有 Tensorflow 2.1.0.
这是答案 I am not using CUDA or anything
。在 CPU 上的训练比在 GPU 上慢得多。如果你没有足够高性能的显卡,你可以使用一些服务,例如 Google Colab 或 Kaggle