Suppot entity parameter

[YevheniiSemendiak]

From W&B docs https://docs.wandb.ai/ref/python/init:

(str, optional) An entity is a username or team name where you're sending runs. This entity must exist before you can send runs there, so make sure to create your account or team in the UI before starting to log runs. If you don't specify an entity, the run will be sent to your default entity, usually your username. Change your default entity in your settings under "default location to create new projects".

If none is specified, the default one will be used which could be inconvenient/undesired behavior.