-
Notifications
You must be signed in to change notification settings - Fork 7
Description
Hello,I'm a beginner who has just started to explore this field,and I get an error when running the following code:
# text-to-motion
bash ./tools/single_test.sh
ModuleNotFoundError: TextMotionDataset: T2MContrastiveModel_SMPLX: No module named 'tma'
However, when I visited https://github.com/IDEA-Research/HumanTOMATO/tree/main/OpenTMA/tma, there has no module called:https://github.com/IDEA-Research/HumanTOMATO/tree/main/OpenTMA/tma
All error messages are below:
$ bash ./tools/single_test.sh
1499it [00:00, 5640.02it/s]
Traceback (most recent call last):
File "/usr/local/miniconda/envs/mc4/lib/python3.9/site-packages/mmcv/utils/registry.py", line 69, in build_from_cfg
return obj_cls(**args)
File "/home/hh/PycharmProjects/MotionCraft/MotionCraft-main/mogen/models/rnns/t2m_bigru_smplx.py", line 404, in init
self.load_pretrained(init_cfg['checkpoint'])
File "/home/hh/PycharmProjects/MotionCraft/MotionCraft-main/mogen/models/rnns/t2m_bigru_smplx.py", line 421, in load_pretrained
state_dict = torch.load(ckpt_path, map_location="cpu")["state_dict"]
File "/usr/local/miniconda/envs/mc4/lib/python3.9/site-packages/torch/serialization.py", line 1097, in load
return _load(
File "/usr/local/miniconda/envs/mc4/lib/python3.9/site-packages/torch/serialization.py", line 1525, in _load
result = unpickler.load()
File "/usr/local/miniconda/envs/mc4/lib/python3.9/site-packages/torch/serialization.py", line 1515, in find_class
return super().find_class(mod_name, name)
ModuleNotFoundError: No module named 'tma'
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/usr/local/miniconda/envs/mc4/lib/python3.9/site-packages/mmcv/utils/registry.py", line 69, in build_from_cfg
return obj_cls(**args)
File "/home/hh/PycharmProjects/MotionCraft/MotionCraft-main/mogen/datasets/text_motion_dataset.py", line 50, in init
super(TextMotionDataset, self).init(data_prefix=data_prefix,
File "/home/hh/PycharmProjects/MotionCraft/MotionCraft-main/mogen/datasets/base_dataset.py", line 57, in init
self.prepare_evaluation()
File "/home/hh/PycharmProjects/MotionCraft/MotionCraft-main/mogen/datasets/base_dataset.py", line 102, in prepare_evaluation
self.evaluator_model = build_submodule(
File "/home/hh/PycharmProjects/MotionCraft/MotionCraft-main/mogen/models/builder.py", line 31, in build_submodule
return SUBMODULES.build(cfg)
File "/usr/local/miniconda/envs/mc4/lib/python3.9/site-packages/mmcv/utils/registry.py", line 237, in build
return self.build_func(*args, **kwargs, registry=self)
File "/home/hh/PycharmProjects/MotionCraft/MotionCraft-main/mogen/models/builder.py", line 8, in build_from_cfg
return MMCV_MODELS.build_func(cfg, registry, default_args)
File "/usr/local/miniconda/envs/mc4/lib/python3.9/site-packages/mmcv/cnn/builder.py", line 27, in build_model_from_cfg
return build_from_cfg(cfg, registry, default_args)
File "/usr/local/miniconda/envs/mc4/lib/python3.9/site-packages/mmcv/utils/registry.py", line 72, in build_from_cfg
raise type(e)(f'{obj_cls.name}: {e}')
ModuleNotFoundError: T2MContrastiveModel_SMPLX: No module named 'tma'
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/home/hh/PycharmProjects/MotionCraft/MotionCraft-main/./tools/test.py", line 128, in
main()
File "/home/hh/PycharmProjects/MotionCraft/MotionCraft-main/./tools/test.py", line 82, in main
dataset = build_dataset(cfg.data.test)
File "/home/hh/PycharmProjects/MotionCraft/MotionCraft-main/mogen/datasets/builder.py", line 38, in build_dataset
dataset = build_from_cfg(cfg, DATASETS, default_args)
File "/usr/local/miniconda/envs/mc4/lib/python3.9/site-packages/mmcv/utils/registry.py", line 72, in build_from_cfg
raise type(e)(f'{obj_cls.name}: {e}')
ModuleNotFoundError: TextMotionDataset: T2MContrastiveModel_SMPLX: No module named 'tma'
Thank you very much for your help if possible.