(self, check_idx)
| 372 | pl_module.train() |
| 373 | |
| 374 | def check_frequency(self, check_idx): |
| 375 | if ((check_idx % self.batch_freq) == 0 or (check_idx in self.log_steps)) and ( |
| 376 | check_idx > 0 or self.log_first_step): |
| 377 | try: |
| 378 | self.log_steps.pop(0) |
| 379 | except IndexError as e: |
| 380 | print(e) |
| 381 | pass |
| 382 | return True |
| 383 | return False |
| 384 | |
| 385 | def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, dataloader_idx): |
| 386 | if not self.disabled and (pl_module.global_step > 0 or self.log_first_step): |