Is there any other method to plot the training and validation losses since the tensorboard is not working in Modulus?
1 Like
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.