keras模型保存错误

Question

-在ubuntu16.04.1 - 在喀拉斯 2.1.6

这是我的尝试

model =Sequential()
model.add(Dense(4096, input_shape=(3,), activation='relu'))
model.add(Dense(2048, activation='relu'))
model.add(Dense(1024, activation='relu'))
model.add(Dense(512, activation='relu'))
model.add(Dense(256, activation='relu'))
model.add(Dense(128, activation='relu'))
model.add(Dense(64, activation='relu'))
model.add(Dense(32, activation='relu'))
model.add(Dense(16, activation='relu'))
model.add(Dense(8, activation='relu'))
model.add(Dense(4, activation='relu'))
model.add(Dense(2, activation='relu'))
model.add(Dense(1, activation='sigmoid'))
parallel_model = multi_gpu_model(model, gpus=4)
parallel_model.compile(loss='binary_crossentropy', optimizer = 'adam', metrics = ['accuracy'])
parallel_model.fit(x_train,y_train,epochs = 1, batch_size =2000)
model.save('test_train_model_re_re.h5')

保存一下，就出现了这个错误。

TypeError：无法 pickle NotImplementedType 对象

我没有使用 call_back 方法。

为什么会出现这个错误，请给出一些解决方案

Answer 1

我也遇到了同样的错误，我解决这个问题的方法是在您的模型适合后使用此代码。

parallel_model.fit(x_train,y_train,epochs = 1, batch_size =2000) 

from keras.models import model_from_json   
# serialize model to JSON
model_json = parallel_model.to_json()
with open("model.json", "w") as json_file:
    json_file.write(model_json)
# serialize weights to HDF5
model.save_weights("model.h5")
print("Saved model to disk")

keras模型保存错误

keras model saving error

python

pickle

keras