在多核上运行lstm时,Python脚本挂起

时间:2018-08-24 09:14:11

标签: python multiprocessing

我正在尝试使用多处理池在不同的数据上创建不同的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,则程序可以工作。代码有什么问题?

0 个答案:

没有答案