From 681bef80aa50a7e712993e11e42c1a91e9af0233 Mon Sep 17 00:00:00 2001 From: Chuanbo Hua Date: Thu, 19 Sep 2024 00:44:06 +0900 Subject: [PATCH] [BugFix] Fix PCTSP stochastic prize generation distribution problem --- rl4co/envs/routing/pctsp/generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rl4co/envs/routing/pctsp/generator.py b/rl4co/envs/routing/pctsp/generator.py index 9ff58fa8..332a5c9c 100644 --- a/rl4co/envs/routing/pctsp/generator.py +++ b/rl4co/envs/routing/pctsp/generator.py @@ -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