(self)
| 57 | self.log_dict(dico, sync_dist=True, rank_zero_only=True) |
| 58 | |
| 59 | def on_test_epoch_end(self): |
| 60 | # Log metrics |
| 61 | dico = self.metrics_log_dict() |
| 62 | # Write to log only if not sanity check |
| 63 | if not self.trainer.sanity_checking: |
| 64 | self.log_dict(dico, sync_dist=True, rank_zero_only=True) |
| 65 | self.save_npy(self.test_step_outputs) |
| 66 | self.rep_i = self.rep_i + 1 |
| 67 | # Free up the memory |
| 68 | self.test_step_outputs.clear() |
| 69 | |
| 70 | def preprocess_state_dict(self, state_dict): |
| 71 | new_state_dict = OrderedDict() |
nothing calls this directly
no test coverage detected