Tensorboard logging is happening per step instead of per epoch

This is the current result for a 20-epoch training in my machine:

The "epoch" number is not showing on this, as you can see, but rather the "step" (?). I also don't know how to make this looks nicer, eg by combining both train/valid losses and make the plots more colourful.

mentioned in issue #8 (closed)

It seems we have to assign the number of epochs to the step key in the metrics dict:

https://github.com/Lightning-AI/lightning/issues/2110

Or we create a custom logger, as discussed in #8 (closed)

mentioned in commit 026ddb48

I managed to get this working on our working branch!

mentioned in commit 64fda27e

mentioned in commit adf29dd1

closed with merge request !6 (merged)