(self, check_idx)
| 81 | pl_module.train() |
| 82 | |
| 83 | def check_frequency(self, check_idx): |
| 84 | return check_idx % self.batch_freq == 0 |
| 85 | |
| 86 | # def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, dataloader_idx): |
| 87 | def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx): |