recursionpharma / gflownet

GFlowNet library specialized for graph & molecular data
MIT License
216 stars 43 forks source link

Separate P_B trunk & min_len #107

Open bengioe opened 1 year ago

bengioe commented 1 year ago

This PR makes it possible to have a separate set of parameters for P_B by setting cfg.model.do_separate_p_b. It also introduces cfg.algo.min_len, which defaults to 0 and prevents and agent from stopping before min_len steps have been taken.

Other changes:

bengioe commented 1 year ago

Basic experiments on the double trunk model (2pPB) vs shared parameterized (pPB) vs uniform (uPB). The other parameter is sampling_tau image

image