Я работаю над обучением модели VGG16 в Керасе. Я использую набор данных TAM и получил эту ошибку. Я прочитал несколько похожих вопросов, но ни одна из них мне не помогла.
Вот код
for layer in model.layers:
layer.trainable = True
model.compile(optimizer=optimizers.SGD(lr=0.0001, momentum=0.9),
loss='categorical_crossentropy', metrics = ['accuracy'])
checkpoint = ModelCheckpoint(filepath=model_weights, save_best_only=False, verbose=0)
reduce_lr = ReduceLROnPlateau(monitor='loss', factor=0.2, patience=10, min_lr=0.00001)
history = model.fit_generator(
generator=train_batches,
steps_per_epoch=10,
epochs=5,
callbacks=[checkpoint, reduce_lr],
validation_data=valid_batches,
validation_steps=100,
verbose=1
)