Method
__init__
(self, resume, now, logdir, ckptdir, cfgdir, config, lightning_config)
Source from the content-addressed store, hash-verified
| 442 | |
| 443 | class SetupCallback(Callback): |
| 444 | def __init__(self, resume, now, logdir, ckptdir, cfgdir, config, lightning_config): |
| 445 | super().__init__() |
| 446 | self.resume = resume |
| 447 | self.now = now |
| 448 | self.logdir = logdir |
| 449 | self.ckptdir = ckptdir |
| 450 | self.cfgdir = cfgdir |
| 451 | self.config = config |
| 452 | self.lightning_config = lightning_config |
| 453 | |
| 454 | def on_keyboard_interrupt(self, trainer, pl_module): |
| 455 | if trainer.global_rank == 0: |
Callers
nothing calls this directly
Tested by
no test coverage detected