search

GFNOrg / torchgfn

A modular, easy to extend GFlowNet library
https://torchgfn.readthedocs.io/en/latest/
Other
231 stars 30 forks source link

Implement the Flow Matching loss #2

Closed saleml closed 2 years ago

saleml commented 2 years ago

Then try it on the HyperGrid environment (make the loss accessible via the configs for train.py). Make sure to reproduce the results reported in Trajectory Balance: Improved Credit Assignment in GFlowNets.

danjenson commented 2 years ago

Here is an implementation that may help, although my abstractions are slightly different (i.e. I have a BatchFlowStep class): https://github.com/ThrunGroup/gfn/blob/main/gfn/losses.py