Skip to content

Commit

Permalink
fix: seh-rewards were not scaled down in seh_freg_moo -- removed flat…
Browse files Browse the repository at this point in the history
…_reward_transform from SehFragMOO to use flat_reward_transform() from the single-objective task
  • Loading branch information
julienroyd committed Feb 27, 2024
1 parent 32e9988 commit 97a459a
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions src/gflownet/tasks/seh_frag_moo.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,9 +88,6 @@ def __init__(
)
assert set(self.objectives) <= {"seh", "qed", "sa", "mw"} and len(self.objectives) == len(set(self.objectives))

def flat_reward_transform(self, y: Union[float, Tensor]) -> FlatRewards:
return FlatRewards(torch.as_tensor(y))

def inverse_flat_reward_transform(self, rp):
return rp

Expand Down

0 comments on commit 97a459a

Please sign in to comment.