如何修复 Tensorflow InvalidArgumentError

Question

这是我的 model

model = tf.keras.models.Sequential([
tf.keras.layers.Conv2D(32, (3,3), activation='relu', input_shape=(150, 150, 3)),
tf.keras.layers.MaxPooling2D(2, 2),
tf.keras.layers.Conv2D(64, (3,3), activation='relu'),
tf.keras.layers.MaxPooling2D(2,2),
tf.keras.layers.Conv2D(128, (3,3), activation='relu'),
tf.keras.layers.MaxPooling2D(2,2),
tf.keras.layers.Conv2D(256, (3,3), activation='relu'),
tf.keras.layers.MaxPooling2D(2,2),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(512, activation='relu'),
tf.keras.layers.Dense(1, activation='softmax') )

Model 适合：

testing = model.fit(
  train_generator,
  steps_per_epoch=25,
  epochs=20,
  validation_data=validation_generator,
  validation_steps=5,
  verbose=1)

错误：

InvalidArgumentError:  Matrix size-incompatible: In[0]: [32,3], In[1]: [512,1]
     [[node gradient_tape/sequential/dense_1/MatMul (defined at <ipython-input-11-34d2a6f3254c>:7) ]] [Op:__inference_train_function_935]


Function call stack:
train_function

我的训练形状是 1312 张图片

我找不到错误任何人都可以帮我解释如何解决它？

Answer 1

补丁大小不正确。 请与下面的讨论 go。

https://github.com/tensorlayer/srgan/issues/152