(self, phase, loss, loss_kl, **kwargs)
| 15 | |
| 16 | |
| 17 | def after_forward_pass(self, phase, loss, loss_kl, **kwargs): |
| 18 | |
| 19 | # / act as seperators. If you would like to log train and test separately you would log test loss in test/loss |
| 20 | wandb.log({'train/loss': loss}, step=phase.iter_nr) |
| 21 | if loss_kl!=0: |
| 22 | wandb.log({'train/loss_kl': loss_kl}, step=phase.iter_nr) |
| 23 | |
| 24 | |
| 25 | def epoch_ended(self, phase, **kwargs): |
nothing calls this directly
no outgoing calls
no test coverage detected