TensorFlow KeyError Input_1

Question

我正在尝试对Tensorflow MobileNetV2进行再培训，但是当我在以下代码下运行时，我得到了Keyerror Input_1

for test_image, test_label in train_dataset.take(1):break

print(base_model(test_image).shape)

Error Log

非常感谢任何解决方案。谢谢

Answer 1

在输入张量中，添加name属性并将其设置为Input_1。