Skip to content

Commit

Permalink
added back in default recomputing behaviour for pf in off policy mode.
Browse files Browse the repository at this point in the history
  • Loading branch information
josephdviviano committed Nov 21, 2023
1 parent 1acfcce commit e052c82
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/gfn/gflownet/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,6 +152,8 @@ def get_pfs_and_pbs(
if not isinstance(estimator_outputs, type(None)):
idx = torch.ones(trajectories.actions.batch_shape).bool()
estimator_outputs = estimator_outputs[idx]
else:
estimator_outputs = self.pf(valid_states)

# TODO: Remove This is left here to compare the recomputed values with the
# carried forward values -- which strangely don't always seem to
Expand Down

0 comments on commit e052c82

Please sign in to comment.