Log config to wandb

Although there is WandbLoggerHook, but the config of the run is not logged to wandb. Will be nice if this can be done by default as long as WandbLoggerHook is invoked.

Currently, this is what I’m adding to my training script in order to log the config:

wandb_kwargs = [ h['init_kwargs'] for h in cfg.log_config.hooks if h['type'] == 'WandbLoggerHook' ][0]
wandb_kwargs['config'] = cfg._cfg_dict.to_dict()
wandb.init(**wandb_kwargs)
logger.info('Logged cfg to wandb.') 

Is there any way to get access to the cfg object from WandbLoggerHook? I’ll be happy to put in a PR for this if needed.

Thank you!

[ayulockin]

I am trying to achieve the same thing. While investigating, I realized that the WandbLoggerHook methods have access to the runner object. However, the runner has no cfg class variable.

If the constructor of BaseRunner can accept cfg then the WandbLoggerHook (and other hooks) can use it to log the config file, something like this:

def before_run(self, runner):
        super(WandbLogger, self).before_run(runner)
        self.wandb.config = runner.cfg

Something as simple as this, should do the trick.

class BaseRunner(metaclass=ABCMeta):
       ....
      def __init__(self, ...., cfg=None):
           self.cfg = cfg

In the train.py we can then do runner.cfg = cfg.

Thoughts? If this is an acceptable solution, I can make a PR.

[ayulockin]

Hey @zhouzaida, would love to know what you think about it. I can make a PR if you would like.