Hello, I am running the CEVAE tutorial Example: Causal Effect VAE — Pyro Tutorials 1.7.0 documentation on Pyro 1.5.2 and I would like to change the way the losses are printed. I am using:
I am assuming that this handles how CEVAE outputs the losses when calling fit. Is there a way to print alter this, for instance showing the loss every 10 epochs? Also I would like to know how I can switch it on and off, currently I am restarting the kernel and not calling the loggers…
Thank you in advance!