KeyError: 'intent_vec'

[SUPER-ALEX]

When I run： python scripts/dump_grabnet_gen.py -b 1 -g 0 --split test --exp_id eval_Intgen --cfg config/oishape_bm/intent/test_GrabIntentNet_OIShape_trigger.yml

error: 2024-05-23 00:24:58,552 [ INFO] => Loading CoarseIntentNetIntentEmbedEncode pretrained model from: checkpoints/intent/trigger_sprayer/CoarseIntentNetIntentEmbedEncode.pth.tar 2024-05-23 00:24:58,649 [ INFO] => Loading SUCCEEDED 2024-05-23 00:24:58,833 [ INFO] => Loading RefineNet pretrained model from: checkpoints/grabnet_oishape/RefineNet.pth.tar 2024-05-23 00:24:58,841 [ INFO] => Loading SUCCEEDED test Epoch 0: 0%| | 0/42 [00:02<?, ?it/s] Traceback (most recent call last): File "/root/OakInk-Grasp-Generation/scripts/dump_grabnet_gen.py", line 109, in main(cfg, arg, exp_time) File "/root/OakInk-Grasp-Generation/scripts/dump_grabnetgen.py", line 84, in main prd, = model(inp=inp, step_idx=i, mode="test", callback=dumper) # GrabNet, HoverGen File "/root/miniconda3/envs/oishape_bm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, kwargs) File "/root/miniconda3/envs/oishape_bm/lib/python3.10/site-packages/torch/nn/parallel/data_parallel.py", line 169, in forward return self.module(*inputs[0], *kwargs[0]) File "/root/miniconda3/envs/oishape_bm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(input, kwargs) File "/root/OakInk-Grasp-Generation/lib/models/grabnet/grabnet_arch.py", line 27, in forward return self.testing_step(inp, step_idx, kwargs) File "/root/OakInk-Grasp-Generation/lib/models/grabnet/grabnet_arch.py", line 34, in testing_step cnetres, = self.CNet(inp=inp, mode='test', step_idx=step_idx, kwargs) File "/root/miniconda3/envs/oishape_bm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, kwargs) File "/root/OakInk-Grasp-Generation/lib/models/grabnet/coarse_net.py", line 85, in forward return self.testing_step(inp, step_idx, kwargs) File "/root/OakInk-Grasp-Generation/lib/models/grabnet/coarse_intent_net.py", line 132, in testing_step intent_vec = inp[Queries.INTENT_VEC] KeyError: 'intent_vec'