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