sweep_id = wandb.sweep(sweep_config,project="testing_sweep2")
wandb.agent(sweep_id, function=train_models)

        __
        __