FangyunWei / SLRT

228 stars 43 forks source link

Keypoints #15

Open twh20151664 opened 1 year ago

twh20151664 commented 1 year ago

When I run the following command to extract keypoints: config_file=configs/rgb_frame64.yaml' python -m torch.distributed.launch --nproc_per_node 4 --master_port 29999 --use_env gen_pose.py --config=${config_file} --split=train the error occurred:‘’File "gen_pose.py", line 214, in main assert pose_results.shape == (batch_data['vlens'][0], 133, 3) AssertionError ERROR:torch.distributed.elastic.multiprocessing.api:failed (exitcode: 1) local_rank: 0 (pid: 55436) of binary"I need you help

2000ZRL commented 1 year ago

Please check your dataloader, batch_data['vlens'] should be a list.