alexhernandezgarcia / gflownet

Generative Flow Networks - GFlowNet
https://gflownet.readthedocs.io/en/latest/
Apache License 2.0
167 stars 11 forks source link

Detailed Balance loss #256

Closed alexhernandezgarcia closed 10 months ago

alexhernandezgarcia commented 10 months ago

This PR implements the detailed balance objective, building upon some of the code developed for the forward-looking loss.

Experiments with the Grid and the Tetris on wandb: https://wandb.ai/alexhg/detailedbalance