Skip to content

Commit 0fbda89

Browse files
authored
Fix for noise sampling for the TD3 exploration (#260)
1 parent 1878a4f commit 0fbda89

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cleanrl/td3_continuous_action.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ def forward(self, x):
178178
else:
179179
with torch.no_grad():
180180
actions = actor(torch.Tensor(obs).to(device))
181-
actions += torch.normal(actor.action_bias, actor.action_scale * args.exploration_noise)
181+
actions += torch.normal(0, actor.action_scale * args.exploration_noise)
182182
actions = actions.cpu().numpy().clip(envs.single_action_space.low, envs.single_action_space.high)
183183

184184
# TRY NOT TO MODIFY: execute the game and log data.

0 commit comments

Comments
 (0)