diff --git a/README.md b/README.md index 12af5096..627f933f 100644 --- a/README.md +++ b/README.md @@ -199,4 +199,4 @@ Currently, the implemented losses are: - Log Partition Variance loss. Introduced [here](https://arxiv.org/abs/2302.05446) # Scripts -Example scripts are provided [here](https://github.com/saleml/torchgfn/tree/master/tutorials/scripts/). They can be used to reproduce published results in the HyperGrid environment, and the Box environment. \ No newline at end of file +Example scripts are provided [here](https://github.com/saleml/torchgfn/tree/master/tutorials/examples/). They can be used to reproduce published results in the HyperGrid environment, and the Box environment. \ No newline at end of file diff --git a/docs/source/conf.py b/docs/source/conf.py index 44c31583..eda470d4 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -9,8 +9,8 @@ import sys project = "torchgfn" -copyright = "2022, Salem Lahlou" -author = "Salem Lahlou" +copyright = "2022-2023, Salem Lahlou, Joseph Viviano, & Victor Schmidt" +author = "Salem Lahlou, Joseph Viviano, & Victor Schmidt" sys.path.insert(0, os.path.abspath("../..")) @@ -27,8 +27,15 @@ "autoapi.extension", "sphinx.ext.napoleon", ] + +source_suffix = { + ".rst": "restructuredtext", + ".txt": "restructuredtext", + ".md": "markdown", +} + autoapi_type = "python" -autoapi_dirs = ["../../src/gfn"] +autoapi_dirs = ["../../src/gfn", "../../tutorials"] autoapi_member_order = "alphabetical" autodoc_typehints = "description" diff --git a/tutorials/README.md b/tutorials/README.md index 7d8a0c51..90e05758 100644 --- a/tutorials/README.md +++ b/tutorials/README.md @@ -1,5 +1,5 @@ # Welcome to the `torchgfn` tutorials 1. Learn the building blocks of your GflowNet with [notebooks](https://github.com/saleml/torchgfn/tree/master/tutorials/notebooks/) -2. See `torchgfn` in action with example [training scripts](https://github.com/saleml/torchgfn/tree/master/tutorials/scripts/) +2. See `torchgfn` in action with example [training scripts](https://github.com/saleml/torchgfn/tree/master/tutorials/examples/) 3. Read a summary of what you need to do to create your own [Environment](https://github.com/saleml/torchgfn/tree/master/tutorials/ENV.md) diff --git a/tutorials/scripts/README.md b/tutorials/examples/README.md similarity index 100% rename from tutorials/scripts/README.md rename to tutorials/examples/README.md diff --git a/tutorials/scripts/__init__.py b/tutorials/examples/__init__.py similarity index 100% rename from tutorials/scripts/__init__.py rename to tutorials/examples/__init__.py diff --git a/tutorials/scripts/train_box.py b/tutorials/examples/train_box.py similarity index 99% rename from tutorials/scripts/train_box.py rename to tutorials/examples/train_box.py index d3762a13..30d6d854 100644 --- a/tutorials/scripts/train_box.py +++ b/tutorials/examples/train_box.py @@ -1,4 +1,4 @@ -""" +r""" The goal of this script is to reproduce some of the published results on the Box environment. Run one of the following commands to reproduce some of the results in [A theory of continuous generative flow networks](https://arxiv.org/abs/2301.12594) diff --git a/tutorials/scripts/train_discreteebm.py b/tutorials/examples/train_discreteebm.py similarity index 99% rename from tutorials/scripts/train_discreteebm.py rename to tutorials/examples/train_discreteebm.py index 6a713615..84191f2d 100644 --- a/tutorials/scripts/train_discreteebm.py +++ b/tutorials/examples/train_discreteebm.py @@ -1,4 +1,4 @@ -""" +r""" The goal of this script is to reproduce some of the published results on the HyperGrid environment. Run one of the following commands to reproduce some of the results in [Trajectory balance: Improved credit assignment in GFlowNets](https://arxiv.org/abs/2201.13259) diff --git a/tutorials/scripts/train_hypergrid.py b/tutorials/examples/train_hypergrid.py similarity index 99% rename from tutorials/scripts/train_hypergrid.py rename to tutorials/examples/train_hypergrid.py index 7acf5599..5a613faf 100644 --- a/tutorials/scripts/train_hypergrid.py +++ b/tutorials/examples/train_hypergrid.py @@ -1,4 +1,4 @@ -""" +r""" The goal of this script is to reproduce some of the published results on the HyperGrid environment. Run one of the following commands to reproduce some of the results in [Trajectory balance: Improved credit assignment in GFlowNets](https://arxiv.org/abs/2201.13259)