是否可以通过 CPU 多处理来拟合 Keras 模型？答案

【问题标题】：Is it possible to fit a Keras model via CPU multiprocessing?是否可以通过 CPU 多处理来拟合 Keras 模型？
【发布时间】：2020-09-04 14:35:46
【问题描述】：

是否可以通过 4 个 CPU 上的多处理来拟合 Keras 模型？（没有 GPU）如果是这样，如何处理？

【问题讨论】：

标签： machine-learning keras multiprocessing python-multiprocessing model-fitting

【解决方案1】：

默认情况下，Keras 将尝试使用您机器上所有可用的内核并行（多处理）拟合您的模型。
您可以使用这个 sn-p 选择 cpu（或作业）的数量：

from keras import backend as K
import tensorflow as tf
jobs = 4 # number of cores
config = tf.ConfigProto(intra_op_parallelism_threads=jobs,
                     inter_op_parallelism_threads=jobs,
                     allow_soft_placement=True,
                     device_count={'CPU': jobs})
session = tf.Session(config=config)
K.set_session(session)

【讨论】：

`Thx，我使用 model = keras...Sequential()。如何将其集成到 model.fit() 函数中？
代码 sn-p 应该适用于您使用的任何 keras api。刚过去程序开始处的 sn-p 代码，你就可以开始了:)