Skip to content

Commit

Permalink
add setting device=cuda in the readme
Browse files Browse the repository at this point in the history
  • Loading branch information
@engmubarak48
engmubarak48 committed Jun 11, 2024
1 parent cff3cc5 commit 92bf362
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -205,6 +205,8 @@ The above command will overwrite the `env` and `proxy` default configuration wit

Hydra configuration is hierarchical. For instance, You can seamlessly modify exisiting flag or variable in the configuration by setting `logger.do.online=False`. For more, feel free to read the [Hydra documentation](https://hydra.cc/docs/intro/).

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`.

## GFlowNet loss functions

Currently, the implementation includes the following GFlowNet losses:
Expand Down

0 comments on commit 92bf362

Please sign in to comment.