W&B: expose "entity" argument

To log a training with W&B, one needs to log in (using wandb login command in terminal and providing an API key/token), specify the project name (for this, we pass the dataset name), and optionally the entity (which is a username or a team name). Currently we don't use the entity argument. We need to add and expose it at the level of the model.fit() method.