Method
save_config
(
self, trainer: Trainer, pl_module: LightningModule, stage: str
)
Source from the content-addressed store, hash-verified
| 46 | """ |
| 47 | |
| 48 | def save_config( |
| 49 | self, trainer: Trainer, pl_module: LightningModule, stage: str |
| 50 | ) -> None: |
| 51 | for logger in trainer.loggers: |
| 52 | if issubclass(type(logger), WandbLogger): |
| 53 | logger.experiment.config.update(self.config.as_dict()) |
| 54 | return super().save_config(trainer, pl_module, stage) |
| 55 | |
| 56 | |
| 57 | class PlotsCallback(Callback): |
Callers
nothing calls this directly
Tested by
no test coverage detected