hongshaorou/tdmpc2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: handle _action_masks buffer in single-task scenarios (#67)

Browse Source
This commit is contained in:
Sue Hyun Park
2025-05-20 12:12:12 +09:00
committed by GitHub
parent 7992fa193e
commit 8bbc14ebab
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tdmpc2/common/layers.py
View File

@@ -212,7 +212,8 @@ def api_model_conversion(target_state_dict, source_state_dict):
# copy log_std_min and log_std_max from target_state_dict to new_state_dict # copy log_std_min and log_std_max from target_state_dict to new_state_dict
new_state_dict['log_std_min'] = target_state_dict['log_std_min'] new_state_dict['log_std_min'] = target_state_dict['log_std_min']
new_state_dict['log_std_dif'] = target_state_dict['log_std_dif'] new_state_dict['log_std_dif'] = target_state_dict['log_std_dif']
new_state_dict['_action_masks'] = target_state_dict['_action_masks'] if '_action_masks' in target_state_dict:
new_state_dict['_action_masks'] = target_state_dict['_action_masks']
# copy new_state_dict to source_state_dict # copy new_state_dict to source_state_dict
source_state_dict.update(new_state_dict) source_state_dict.update(new_state_dict)