如何修復 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