Computation-aware Bayesian optimization

Please install libraries in the requirements.txt file using the following command:

pip install -r requirements.txt

Use of virtual environments is optional but recommended. For GPU support refer to the PyTorch docs to ensure correct the CUDA version.

Also be sure to specify/change the save_dir and data_dir configs in the main.py.