- First of all ensure the data is downloaded by running the bash script
get_data.sh
in 'scripts/`. - Move the data to the main repo as
mv data_local ../
- Install all dependencies as
pip install -r requirements.txt
. - Then, for running the parameter sweep, do:
wandb sweep sweep.yaml
, which will give you a sweep_id (actually a command as wandb agent ...) - Run the command
wandb agent <sweep_id>
Reference for the sweeps: https://github.com/borisdayma/lightning-kitti.
The data and model checkpoints are in "geo2020_supermag/Uniformdata_multiyear/" in the bucket.