tutorials.examples.train_hypergrid_simple ========================================= .. py:module:: tutorials.examples.train_hypergrid_simple .. autoapi-nested-parse:: A simplified version of GFlowNet training on the HyperGrid environment, focusing on the core concepts. This script implements Trajectory Balance (TB) training with minimal features to aid understanding. Example usage: python train_hypergrid_simple.py --ndim 2 --height 8 --epsilon 0.1 Key differences from the full version: - Only implements TB loss - No replay buffer - No wandb integration - Simpler architecture with shared trunks - Basic command line options Attributes ---------- .. autoapisummary:: tutorials.examples.train_hypergrid_simple.parser Functions --------- .. autoapisummary:: tutorials.examples.train_hypergrid_simple.main Module Contents --------------- .. py:function:: main(args) .. py:data:: parser