(self, check_idx)
| 597 | pl_module.train() |
| 598 | |
| 599 | def check_frequency(self, check_idx): |
| 600 | if ((check_idx % self.batch_freq) == 0 or (check_idx in self.log_steps)) and ( |
| 601 | check_idx > 0 or self.log_first_step): |
| 602 | try: |
| 603 | self.log_steps.pop(0) |
| 604 | except IndexError as e: |
| 605 | print(e) |
| 606 | pass |
| 607 | return True |
| 608 | return False |
| 609 | |
| 610 | def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, dataloader_idx): |
| 611 | if not self.disabled and (pl_module.global_step > 0 or self.log_first_step): |