GuyTevet / motion-diffusion-model

The official PyTorch implementation of the paper "Human Motion Diffusion Model"
MIT License
3.12k stars 337 forks source link

Train error #120

Closed CrazyLearner98 closed 1 year ago

CrazyLearner98 commented 1 year ago

When I trian the HumanML3D for text to motion, I just meet these error, can someone tell me what happend here?

Traceback (most recent call last): File "/home/zonghengli/anaconda3/envs/mdm/lib/python3.7/runpy.py", line 193, in _run_module_as_main "main", mod_spec) File "/home/zonghengli/anaconda3/envs/mdm/lib/python3.7/runpy.py", line 85, in _run_code exec(code, run_globals) File "/home/zonghengli/Study/motion-diffusion-model/train/train_mdm.py", line 49, in main() File "/home/zonghengli/Study/motion-diffusion-model/train/train_mdm.py", line 36, in main data = get_dataset_loader(name=args.dataset, batch_size=args.batch_size, num_frames=args.num_frames) File "/home/zonghengli/Study/motion-diffusion-model/data_loaders/get_data.py", line 44, in get_dataset_loader dataset = get_dataset(name, num_frames, split, hml_mode) File "/home/zonghengli/Study/motion-diffusion-model/data_loaders/get_data.py", line 37, in get_dataset dataset = DATA(split=split, num_frames=num_frames, mode=hml_mode) File "/home/zonghengli/Study/motion-diffusion-model/data_loaders/humanml/data/dataset.py", line 764, in init self.t2m_dataset = Text2MotionDatasetV2(self.opt, self.mean, self.std, self.split_file, self.w_vectorizer) File "/home/zonghengli/Study/motion-diffusion-model/data_loaders/humanml/data/dataset.py", line 275, in init name_list, length_list = zip(*sorted(zip(new_name_list, length_list), key=lambda x: x[1])) ValueError: not enough values to unpack (expected 2, got 0)

Lingxiao108105 commented 1 year ago

I got the same issue when training the KIT dataset. And I manage to solve it according to #90.

Maybe you did not unzip some files in the folder \dataset\HumanML3D ?

CrazyLearner98 commented 1 year ago

I got the same issue when training the KIT dataset. And I manage to solve it according to #90.

Maybe you did not unzip some files in the folder \dataset\HumanML3D ?

Thanks for your reply, I will try the solution in #90, I have two rar files not packing!

GuyTevet commented 1 year ago

Please reopen if this was not resolved by #90