Plotting of training and validation losses

Is there any other method to plot the training and validation losses since the tensorboard is not working in Modulus?

1 Like

Hi @id21resch11019

Presently, no. Tensorboard is our built in logging backend in Modulus. However customizing Modulus to log any way you want is straight forward. In the trainer.py, you can add a logging method of your choice based on the summary_freq. The losses for a given iteration are stored in the losses dictionary.