For installing the environment use setup_mila_cluster.sh
.
A good place to get started is src/tasks/main.py
.
Code for hyperparameter search is in raytune.py
.
To launch the hyperparameter search experiments on the slurm cluster you can use run_exp.sh
.
For more info on the repository structure, see the original Gflownet repository.