I am using the code inside the on_train_epoch_end inside a metric. So the important part is:
` fig = plt.figure()

my plot

logger.experiment.add_figure("fig", fig)
plt.close() `

Let me work on it 👌

GrievingTurkey78 , can you provide a small self contained snippet to run?

GrievingTurkey78 , which script in the repository are you running?

The plot is generated and added to tensorboard but seems clearml is not catching it.