Add "EdgeBasedGFlowNet" and "StateBasedGFlowNet".

GFNOrg / torchgfn

GFlowNet library

https://torchgfn.readthedocs.io/en/latest/

Other

213 stars 26 forks source link

Add "EdgeBasedGFlowNet" and "StateBasedGFlowNet". #119

Closed josephdviviano closed 1 year ago

josephdviviano commented 1 year ago

We have a TrajectoryBasedGFlowNet class, which is used in many other trajectory-based GFNs. It would be nice to abstract out the Edge / State functions into their own class, to help develop new Edge & State-based methods.

saleml commented 1 year ago

https://github.com/saleml/torchgfn/issues/114 duplicate