Method
__init__
(self, resume, now, logdir, ckptdir, cfgdir, config, lightning_config)
Source from the content-addressed store, hash-verified
| 465 | |
| 466 | class SetupCallback(Callback): |
| 467 | def __init__(self, resume, now, logdir, ckptdir, cfgdir, config, lightning_config): |
| 468 | super().__init__() |
| 469 | self.resume = resume |
| 470 | self.now = now |
| 471 | self.logdir = logdir |
| 472 | self.ckptdir = ckptdir |
| 473 | self.cfgdir = cfgdir |
| 474 | self.config = config |
| 475 | self.lightning_config = lightning_config |
| 476 | |
| 477 | def on_keyboard_interrupt(self, trainer, pl_module): |
| 478 | if trainer.global_rank == 0: |
Callers
nothing calls this directly
Tested by
no test coverage detected