From 00cb7e129dadd283619b17fa8d59877317a97fb3 Mon Sep 17 00:00:00 2001 From: alexhernandezgarcia Date: Thu, 6 Jun 2024 09:42:54 -0400 Subject: [PATCH] Add note about device on README --- README.md | 2 ++ 1 file changed, 2 insertions(+) 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