我正在尝试使用多处理池在不同的数据上创建不同的lstm模型。 程序尝试创建lstm层时挂起。
def create_model(neurons, X):
model = Sequential()
model.add(LSTM(neurons, input_shape=(X.shape[1], X.shape[2]), return_sequences=False))
model.add(Dense(1, kernel_initializer='uniform', activation='relu'))
model.compile(loss='mean_squared_error', optimizer='adam', metrics=['mae', 'mse','accuracy'])
return model
from pathos.multiprocessing import ProcessingPool
from itertools import repeat
pool = ProcessingPool(4)
neurons = [50, 100]
results = pool.map(create_model, neurons, repeat(X))
脚本在lstm层创建时挂起。如果我用lstm代替lstm,则程序可以工作。代码有什么问题?