GFNOrg / torchgfn

GFlowNet library
https://torchgfn.readthedocs.io/en/latest/
Other
213 stars 26 forks source link

New abstractions: Add "EdgeBasedGFlowNet" and "StateBasedGFlowNet" #114

Open saleml opened 1 year ago

saleml commented 1 year ago

Similar to TrajectoryBasedGFlowNet, which is the superclass of TB, SubTB, and LogZVar GFlowNets, meaning that the loss function decomposes into a sum of losses over trajectories, we might need to have two more useful abstractions, as mentioned in the Foundations paper: EdgeBasedGFlowNet and StateBasedGFlowNet