(self, check_idx)
| 574 | pl_module.train() |
| 575 | |
| 576 | def check_frequency(self, check_idx): |
| 577 | if ((check_idx % self.batch_freq) == 0 or (check_idx in self.log_steps)) and ( |
| 578 | check_idx > 0 or self.log_first_step): |
| 579 | try: |
| 580 | self.log_steps.pop(0) |
| 581 | except IndexError as e: |
| 582 | print(e) |
| 583 | pass |
| 584 | return True |
| 585 | return False |
| 586 | |
| 587 | def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, dataloader_idx): |
| 588 | if not self.disabled and (pl_module.global_step > 0 or self.log_first_step): |