Method
__init__
(self, resume, now, logdir, ckptdir, cfgdir, config, lightning_config)
Source from the content-addressed store, hash-verified
| 241 | |
| 242 | class SetupCallback(Callback): |
| 243 | def __init__(self, resume, now, logdir, ckptdir, cfgdir, config, lightning_config): |
| 244 | super().__init__() |
| 245 | self.resume = resume |
| 246 | self.now = now |
| 247 | self.logdir = logdir |
| 248 | self.ckptdir = ckptdir |
| 249 | self.cfgdir = cfgdir |
| 250 | self.config = config |
| 251 | self.lightning_config = lightning_config |
| 252 | |
| 253 | def on_keyboard_interrupt(self, trainer, pl_module): |
| 254 | if trainer.global_rank == 0: |
Callers
nothing calls this directly
Tested by
no test coverage detected