Improve Tensorboard Logging
Micky774 opened this issue · comments
In the driver(s) improve tensorboard logging to separately record 3 types of loss:
- Net Loss
- Pseudo Loss
- Sample Loss (gamma=0)
For each of the above losses, we should also record their components: reconstruction and KL divergence.