Проблема размеров, когда продолжить обучение предварительно обученных данных

Question

Я хочу загрузить предварительно обученный файл h5, который я использую для обучения python. Но когда я загружаю файл и пытаюсь тренироваться с той же командой обучения, у него возникает проблема с размерами. Входные измерения - 3, а выходные - 1.

new_model=load_model("model-020-virtuell-test.h5")
score = new_model.evaluate(X_valid, y_valid, verbose=0)
print('Test loss 1:', score[0])
print('Test accuracy 1:', score[1])
#assert_allclose(model.predict(X_train),
#            new_model.predict(X_train),
#            1e-5)
checkpoint = ModelCheckpoint(filepath,
                             monitor='val_loss',
                             verbose=0,
                             save_best_only=True,
                             mode='auto')

history = new_model.fit_generator(batcher(data_dir, X_train, y_train, batch_size, True),
                    samples_per_epoch,
                    nb_epoch,
                    max_q_size=1,
                    validation_data=batcher(data_dir, X_valid, y_valid, batch_size, False),
                    nb_val_samples=len(X_valid),
                    callbacks=[checkpoint],
                    verbose=1)

, и я получил сообщение об ошибке, что лямбда-вход имеет 4 измерения, но получил массив (2786,1).

Обучение без проблем справился с этим кодом.

Я добавил только строку load_model и хотел продолжить обучение этой модели.

Что мне изменить в этом случае?

Спасибо авансовый.