Fix weights and biases. If wandb_mode is not passed in kwargs, then we do not use wandb.