如何强制model.fit方法使用所有cpu。
Tensorflow版本> 1.10
import tensorflow as tf
from tensorflow import keras
model = keras.models.Sequential([
keras.layers.Dense(100, activation=tf.nn.relu, input_shape=(10,)),
keras.layers.Dropout(0.5),
keras.layers.Dense(50, activation=tf.nn.relu),
keras.layers.Dense(10, activation=tf.nn.softmax)
])
model.fit(train_data, target, epochs=100)
任何帮助都是非常有意义的。