To train a model in Tensorflow I checked other sources they use sessions in loop in order to train model.
with tf.Session() as sess:
init.run()
for ep in range(epochs):
sess.run(training_op, feed_dict={X: x_batches, y:y_batches})
mse = loss.eval(feed_dict={X: x_batches, y:y_batches})
Why did you use model.fit?