Skip to content

Commit

Permalink
[BugFix] Fix PCTSP stochastic prize generation distribution problem
Browse files Browse the repository at this point in the history
  • Loading branch information
cbhua committed Sep 18, 2024
1 parent 2812925 commit 681bef8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion rl4co/envs/routing/pctsp/generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ def __init__(
"deterministric_prize", "uniform", 0.0, 4.0 / self.num_loc, **kwargs
)
self.stochastic_prize_sampler = get_sampler(
"stochastic_prize", "uniform", 0.0, 8.0 / self.num_loc, **kwargs
"stochastic_prize", "uniform", 0.0, 2.0, **kwargs
)

# For the penalty to make sense it should be not too large (in which case all nodes will be visited) nor too small
Expand Down

0 comments on commit 681bef8

Please sign in to comment.