when i run the command :
python calm/run.py --test --task HumanoidStrikeFSM --num_envs 16 --cfg_env calm/data/cfg/humanoid_sword_shield_strike_fsm.yaml --cfg_train calm/data/cfg/train/rlg/hrl_humanoid_fsm.yaml --motion_file calm/data/motions/reallusion_sword_shield/dataset_reallusion_sword_shield_fsm_movements.yaml --llc_checkpoint [path_to_llc_checkpoint] --checkpoint [path_to_hlc_checkpoint]
using the pre-trained models
calm/data/models/calm_llc_reallusion_sword_shield.pth
calm/data/models/calm_hlc_precision_trained_reallusion_sword_shield.pth
while the error is
=> loading checkpoint 'calm/data/models/calm_llc_reallusion_sword_shield.pth'
=> loading checkpoint 'calm/data/models/calm_llc_reallusion_sword_shield.pth'
Loaded LLC checkpoint from calm/data/models/calm_llc_reallusion_sword_shield.pth
=> loading checkpoint 'calm/data/models/calm_hlc_precision_trained_reallusion_sword_shield.pth'
Traceback (most recent call last):
File "calm/run.py", line 277, in
main()
File "calm/run.py", line 271, in main
runner.run(vargs)
File "/home/zhouyang/anaconda3/envs/rlgpu/lib/python3.7/site-packages/rl_games/torch_runner.py", line 144, in run
player.run()
File "/home/zhouyang/task_python/CALM/calm/learning/hrl_players.py", line 139, in run
action = self.get_action(obs_dict, is_determenistic)
File "/home/zhouyang/task_python/CALM/calm/learning/hrl_fsm_players.py", line 98, in get_action
clamped_actions[self.env.task._should_strike == 1] = self.env.task._possible_latents[self.env.task._strike_index]
IndexError: The shape of the mask [1] at index 0 does not match the shape of the indexed tensor [64] at index 0
when i run the command : python calm/run.py --test --task HumanoidStrikeFSM --num_envs 16 --cfg_env calm/data/cfg/humanoid_sword_shield_strike_fsm.yaml --cfg_train calm/data/cfg/train/rlg/hrl_humanoid_fsm.yaml --motion_file calm/data/motions/reallusion_sword_shield/dataset_reallusion_sword_shield_fsm_movements.yaml --llc_checkpoint [path_to_llc_checkpoint] --checkpoint [path_to_hlc_checkpoint] using the pre-trained models calm/data/models/calm_llc_reallusion_sword_shield.pth
calm/data/models/calm_hlc_precision_trained_reallusion_sword_shield.pth while the error is
=> loading checkpoint 'calm/data/models/calm_llc_reallusion_sword_shield.pth' => loading checkpoint 'calm/data/models/calm_llc_reallusion_sword_shield.pth' Loaded LLC checkpoint from calm/data/models/calm_llc_reallusion_sword_shield.pth => loading checkpoint 'calm/data/models/calm_hlc_precision_trained_reallusion_sword_shield.pth' Traceback (most recent call last): File "calm/run.py", line 277, in
main()
File "calm/run.py", line 271, in main
runner.run(vargs)
File "/home/zhouyang/anaconda3/envs/rlgpu/lib/python3.7/site-packages/rl_games/torch_runner.py", line 144, in run
player.run()
File "/home/zhouyang/task_python/CALM/calm/learning/hrl_players.py", line 139, in run
action = self.get_action(obs_dict, is_determenistic)
File "/home/zhouyang/task_python/CALM/calm/learning/hrl_fsm_players.py", line 98, in get_action
clamped_actions[self.env.task._should_strike == 1] = self.env.task._possible_latents[self.env.task._strike_index]
IndexError: The shape of the mask [1] at index 0 does not match the shape of the indexed tensor [64] at index 0