(self, phase, **kwargs)
| 39 | |
| 40 | |
| 41 | def epoch_ended(self, phase, **kwargs): |
| 42 | avg_loss_pos=phase.loss_pos_acum_per_epoch/phase.samples_processed_this_epoch |
| 43 | avg_loss_dir=phase.loss_dir_acum_per_epoch/phase.samples_processed_this_epoch |
| 44 | avg_loss_curv=phase.loss_curv_acum_per_epoch/phase.samples_processed_this_epoch |
| 45 | |
| 46 | if phase.grad==False and phase.loss_pos_acum_per_epoch!=0: |
| 47 | self.tensorboard_writer.add_scalar('hair_forge/' + phase.name + '/loss_pos_avg', avg_loss_pos.item(), phase.epoch_nr) |
| 48 | if phase.grad==False and phase.loss_dir_acum_per_epoch!=0: |
| 49 | self.tensorboard_writer.add_scalar('hair_forge/' + phase.name + '/loss_dir_avg', avg_loss_dir.item(), phase.epoch_nr) |
| 50 | if phase.grad==False and phase.loss_curv_acum_per_epoch!=0: |
| 51 | self.tensorboard_writer.add_scalar('hair_forge/' + phase.name + '/loss_curv_avg', avg_loss_curv.item(), phase.epoch_nr) |
| 52 |
nothing calls this directly
no outgoing calls
no test coverage detected