tutorials.examples.train_diffusion_sampler
==========================================

.. py:module:: tutorials.examples.train_diffusion_sampler

.. autoapi-nested-parse::

   Example script for training a diffusion sampler with a GFlowNet loss (Trajectory Balance)
   with SimpleGaussianMixtureTarget as the target unnormalized distribution.

   Here, we use the pinned Brownian motion as the reference process; see https://arxiv.org/abs/2402.05098
   for more details, and see https://arxiv.org/abs/2302.13834 or https://arxiv.org/abs/2211.01364 for
   examples of using the Ornstein-Uhlenbeck process as the reference process.

   Reference: https://github.com/GFNOrg/gfn-diffusion



Attributes
----------

.. autoapisummary::

   tutorials.examples.train_diffusion_sampler.parser


Functions
---------

.. autoapisummary::

   tutorials.examples.train_diffusion_sampler.evaluate_density_metrics
   tutorials.examples.train_diffusion_sampler.main


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

.. py:function:: evaluate_density_metrics(gflownet, env, eval_n, eval_batch_size)

.. py:function:: main(args)

.. py:data:: parser