Еще одна идея: если дело не в пакете то надо попробовать поменять batch size, пример:
history = model.fit(X_train, Y,
validation_data=(X_val, Y_val),
batch_size=3968, # -> пробовал 32, 128, 256, 1024 и т д - скорость обучения увеличилась когда больше 2000 стал
epochs=128,
validation_split=0.15,
callbacks=[early_stopping, reduce_lr, model_chk],
shuffle=False,
verbose=1)
Написано
Войдите на сайт
Чтобы задать вопрос и получить на него квалифицированный ответ.
history = model.fit(X_train, Y,
validation_data=(X_val, Y_val),
batch_size=3968, # -> пробовал 32, 128, 256, 1024 и т д - скорость обучения увеличилась когда больше 2000 стал
epochs=128,
validation_split=0.15,
callbacks=[early_stopping, reduce_lr, model_chk],
shuffle=False,
verbose=1)