Method
__init__
(self, resume, logdir, ckptdir, cfgdir, config)
Source from the content-addressed store, hash-verified
| 87 | |
| 88 | class SetupCallback(Callback): |
| 89 | def __init__(self, resume, logdir, ckptdir, cfgdir, config): |
| 90 | super().__init__() |
| 91 | self.resume = resume |
| 92 | self.logdir = logdir |
| 93 | self.ckptdir = ckptdir |
| 94 | self.cfgdir = cfgdir |
| 95 | self.config = config |
| 96 | |
| 97 | def on_fit_start(self, trainer, pl_module): |
| 98 | if trainer.global_rank == 0: |
Callers
nothing calls this directly
Tested by
no test coverage detected