无法从预先训练的tf.keras模型读取权重

时间:2019-08-13 18:00:31

标签: python tensorflow keras

在加载训练好的模型时,出现以下错误

tensorflow.python.framework.errors_impl.FailedPreconditionError: Error while reading resource variable module_1/InceptionV3/Mixed_6c/Branch_2/Conv2d_0d_7x1/weights from Container: localhost. This could mean that the variable was uninitialized. Not found: Resource localhost/module_1/InceptionV3/Mixed_6c/Branch_2/Conv2d_0d_7x1/weights/class tensorflow::Var does not exist.
     [[{{node lambda_4/module_1_apply_default/InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0d_7x1/Conv2D/ReadVariableOp}}]]

此错误消息是指无法从模型读取的权重。这是我的程序代码

new_model =  tf.keras.models.load_model('box20.h5')
print(new_model.summary())
new_model.predict(test_set)

我在要预测测试集的最后一行收到错误。

我的模型摘要是

  

型号:“ sequential_4”   __________________________________________________________图层(类型)输出形状参数========================================= ============   lambda_4(Lambda)(无,2048)0 _________________________________________________________   density_4(Dense)(None,2)4098
  ================================================== =====总参数:4,098   可调参数:4,098   不可训练的参数:0

     

here提供的解决方案可能能够解决问题。但是我无法很好地理解此解决方案。什么是some_custom_config?

我已经在tf keras版本2.2.4-tf下训练了我的模型,并且正在使用该模型进行加载

0 个答案:

没有答案