train_model.py

print('# Fit model on training data')

history = model.fit(x_train, 
                    x_train, #we pass it the same input data as desired output
                    
                    #If the model is taking forever to train, make this bigger
                    #If it is taking forever to load for the first epoch, make this smaller
                    batch_size=256, 
                    
                    epochs=10,
                    # We pass it validation data to
                    # monitor loss and metrics
                    # at the end of each epoch
                    validation_data=(x_valid, x_valid))