TensorBoard 2.0.0 not updating train scalars

It's probably related to https://github.com/tensorflow/tensorboard/issues/2412.

Adding profile_batch=0 to the callback should solve the problem.