How to avoid using Wandb? I don't have a need for it, but more problems arise here. 
