tutorials.examples.train_discreteebm
====================================

.. py:module:: tutorials.examples.train_discreteebm

.. autoapi-nested-parse::

   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)

   python train_hypergrid.py --ndim 4 --height 8 --R0 {0.1, 0.01, 0.001} --tied {--uniform} --loss {TB, DB}
   python train_hypergrid.py --ndim 2 --height 64 --R0 {0.1, 0.01, 0.001} --tied {--uniform} --loss {TB, DB}

   And run one of the following to reproduce some of the results in
   [Learning GFlowNets from partial episodes for improved convergence and stability](https://arxiv.org/abs/2209.12782)
   python train_hypergrid.py --ndim {2, 4} --height 12 --R0 {1e-3, 1e-4} --tied --loss {TB, DB, SubTB}



Attributes
----------

.. autoapisummary::

   tutorials.examples.train_discreteebm.DEFAULT_SEED
   tutorials.examples.train_discreteebm.parser


Functions
---------

.. autoapisummary::

   tutorials.examples.train_discreteebm.main


Module Contents
---------------

.. py:data:: DEFAULT_SEED
   :value: 4444


.. py:function:: main(args)

.. py:data:: parser