train_complete_model.py

from tensorflow.keras.callbacks import EarlyStopping

early_stop = EarlyStopping(
    monitor='val_loss', min_delta=0, patience=5, verbose=0, mode='auto',
    baseline=None, restore_best_weights=True)

model.fit(x_train, {device:y_train for device in device_names},
          batch_size=64,
          epochs=1000,
          verbose=1,
          validation_data=(x_test, {device:y_test for device in device_names}),callbacks=[early_stop])