我正在使用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]。
。 我该怎么做才能解决问题?