原因はよく分からないが、Sequential APIでなく、Functional APIを使うと直った。 サンプルコードは以下。
base_model = tf.keras.applications.inception_resnet_v2.InceptionResNetV2(weights='imagenet', include_top=False, input_shape=(WIDTH, HEIGHT, 3)) input = Input(shape=(299, 299, 3)) o = base_model(input) o = Flatten()(o) o = Dense(256, activation='relu')(o) o = Dense(8)(o) model = Model(inputs=input, outputs=o)
参考: