[CLI]: WandB to do .watch(model) in lightning cli about wandb

[jkobject]

solved it with :

class MySaveConfig(SaveConfigCallback):
    def __init__(self, *args, wandblog="all", log_freq=500, log_graph=False, **kwargs):
        self.wandblog = wandblog
        self.wandblog_freq = log_freq
        self.wandblog_graph = log_graph
        super().__init__(*args, **kwargs)

    def setup(self, trainer: Trainer, pl_module: LightningModule, stage: str) -> None:
        if type(trainer.logger) is WandbLogger:
            trainer.logger.watch(
                pl_module,
                log=self.wandblog,
                log_freq=self.wandblog_freq,
                log_graph=self.wandblog_graph,
            )
        return super().setup(trainer, pl_module, stage)