Integrating W&B with Lightning CLI

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

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

if __name__ == "__main__":

To launch, I do something like this:

python -u 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?


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!

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.

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!