CPU调用GPU训练的模型

• 如何保存模型 保存模型的整个结构+权重+优化器状态

from keras.models import load_model
model.save('my_model.h5')

• 预测

test_pred1 = load_model1.predict([pro1_vec, pro2_vec])

出现报错： WARNING:tensorflow:5 out of the last 13 calls to <function Model.make_predict_function..p... 将model.predict(X_test)改成model(X_test)

• 当模型带有CUDNN的GRU层 网上有教程说去掉json文件中的cudnn即可，但是我在这一过程中出现了很多报错，可以调用模型的权重然后预测

from tensorflow.keras.models import load_model
# 搭建出模型框架
def build_model():
    ***
    ***
    ***
    return merge_model
model = build_model()
model.load_weights('my_model.h5')
test_pred1 = model([seq1_tensor, seq2_tensor])

参考： 最推荐看这篇： https://blog.csdn.net/weixin_39653948/article/details/105849959

cloud.tencent.com/developer/a… cloud.tencent.com/developer/a… neyzoter.cn/2018/07/14/… blog.csdn.net/luteresa/ar… blog.csdn.net/jclian91/ar… www.icode9.com/content-4-8… zhuanlan.zhihu.com/p/272798504 www.jianshu.com/p/1c7a9e1c3… www.cnblogs.com/tujia/p/138… blog.csdn.net/aaether/art… blog.csdn.net/zhou_438/ar… cloud.tencent.com/developer/a… blog.csdn.net/qq_38978225…