tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add rounding for safety

Browse Source
This commit is contained in:
AdilZouitine
2025-04-08 08:50:02 +00:00
parent a7be613ee8
commit a8135629b4
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/common/policies/sac/modeling_sac.py
View File

@@ -421,6 +421,7 @@ class SACPolicy(
# In the buffer we have the full action space (continuous + discrete) # In the buffer we have the full action space (continuous + discrete)
# We need to split them before concatenating them in the critic forward # We need to split them before concatenating them in the critic forward
actions_discrete: Tensor = actions[:, DISCRETE_DIMENSION_INDEX:].clone() actions_discrete: Tensor = actions[:, DISCRETE_DIMENSION_INDEX:].clone()
actions_discrete = torch.round(actions_discrete)
actions_discrete = actions_discrete.long() actions_discrete = actions_discrete.long()
if complementary_info is not None: if complementary_info is not None: