loggingartificial-intelligencereinforcement-learningwandb

Wandb Sweep tuning define output directory


Please can be defined output directory for wandb when running sweep tuning?

I've defined directory using wandb.init(dir="./out/"), but ./out/ is used only for output of runs, but ./wandb/ folder is created in the directory from which I am running scripts. Can be wandb sweet output directory defined too, so that it uses the same directory as for runs (in my case ./out/) ?

Current final structure after runing script looks like:

./out/wandb/
./wandb

And I want only:

./out/wandb

Solution

  • you can setup the environment variable WANDB_DIR by doing

    export WANDB_DIR=~full_path/out/
    

    I am running the sweep from the terminal by doing:

    > wandb sweep sweep_file.yaml
    > wandb agent sweep_id --count 1
    

    enter image description here

    As you can see, the runs go to the out/ folder