diff --git a/README.md b/README.md index d58256682..a55a0dd7a 100644 --- a/README.md +++ b/README.md @@ -56,6 +56,8 @@ python main.py gflownet=trajectorybalance env=ctorus proxy=torus The above command will overwrite the `env` and `proxy` default configuration with the configuration files in `config/env/ctorus.yaml` and `config/proxy/torus.yaml` respectively. +Note that by default, PyTorch will operate on the CPU because we have not observed performance improvements by running on the GPU. You may run on GPU with `device=cuda`. + Hydra configuration is hierarchical. For instance, a handy variable to change while debugging our code is to avoid logging to wandb. You can do this by setting `logger.do.online=False`. ## GFlowNet loss functions