Integrating W&B with Lightning CLI

[minimario]

I’m using the PyTorch Lightning CLI, with code that looks like this

def main():
    cli = CLI(model class, data module)

if __name__ == "__main__":
    main()

To launch, I do something like this:

python -u main.py fit 

I saw you can integrate it with the PT Lightning trainer, but I’m not sure how to integrate it with the CLI. How can one do this?

Thanks!

[luis_bergua]

Hi @minimario, thanks for writing in! I can see this was requested here and it’s possible with a command like python trainer. py fit --trainer.logger=WandbLogger. Please let me know if this is helpful!

[luis_bergua]

Hi there, I wanted to follow up on this request. Please let us know if we can be of further assistance or if your issue has been resolved.

[luis_bergua]

Hi there, since we have not heard back from you we are going to close this request. If you would like to re-open the conversation, please let us know!