使用keras加载训练模型进行预测时候报错:
'tf' is not defined on load_model()
原因是使用了Lambda自定义层,需要参数声明:
model = load_model('model.model', custom_objects={'tf': tf})
使用keras加载训练模型进行预测时候报错:
'tf' is not defined on load_model()
原因是使用了Lambda自定义层,需要参数声明:
model = load_model('model.model', custom_objects={'tf': tf})