如何解决有关重新加载VGG16模型的Dimension问题?

时间:2018-04-30 16:53:16

标签: python tensorflow keras

我正在使用VGG16模型在猫和狗上制作分类器。我修改了模型的最后一层,有2个节点而不是1000个节点。我在这里保存模型:

model.save(' CNNFromScratch.h5&#39)

并通过以下方式将其加载到另一个文件中:

new_model = load_model(' CNNFromScratch.h5')

我收到以下错误:

  

Traceback(最近一次调用最后一次):文件   " /usr/local/lib/python3.5/site-packages/tensorflow/python/framework/ops.py" ;,   第1567行,在_create_c_op中       c_op = c_api.TF_FinishOperation(op_desc)tensorflow.python.framework.errors_impl.InvalidArgumentError:   两个形状中的尺寸0必须相等,但是为4096和1000。   形状为[4096,2]和[1000,2]。对于“Assign_30' (操作:'分配')   输入形状:[4096,2],[1000,2]。

我该怎么做才能解决问题?

0 个答案:

没有答案