alexhernandezgarcia / gflownet

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

get_rewards vs. get_terminating_rewards in Batch #254

Open alexhernandezgarcia opened 10 months ago

alexhernandezgarcia commented 10 months ago

It may be possible to get rid of get_rewards() altogether since it is probably used by the flow matching loss only and it does not need to get the 0s of the non-terminating states.