PyTorch lightning and MLFLow logging – use epoch number as step
I’m training a DL model using Lightning, this is the general outcome of the module corresponding to the model:
MLFlow & Lightning – Epochs and steps not logging properly
I use Lightning to set up and train my transformer network.
I use MLFlow as a logger.