[TIP] How to connect your WandB run to your team's project

In PyTorch Lightning, in your WandbLogger function, you have to add an `entity` argument. In our case, the function will look like this:
```
wandb_logger = WandbLogger(
        name=wandb_name, entity='ecl-mlstudy', project="STS", offline=args.wandb_offline
    )
```