(self, experiment_name)
| 4 | class TensorboardCallback(Callback): |
| 5 | |
| 6 | def __init__(self, experiment_name): |
| 7 | self.tensorboard_writer = SummaryWriter("tensorboard_logs/"+experiment_name) |
| 8 | self.experiment_name=experiment_name |
| 9 | |
| 10 | |
| 11 | def after_forward_pass(self, phase, loss=0, loss_pos=0, loss_dir=0, loss_curv=0, loss_kl=0, lr=0, z_deviation=None, z=None, z_no_eps=None, **kwargs): |
nothing calls this directly
no outgoing calls
no test coverage detected