Same file root
configs/body_2d_keypoint/associative_embedding/coco/ae_hrnet-w32_8xb24-300e_coco-512x512.py
Reproduces the problem - error message
Traceback (most recent call last):
File "/data/home/seondeok/mmpose/tools/train.py", line 161, in <module>
main()
File "/data/home/seondeok/mmpose/tools/train.py", line 157, in main
runner.train()
File "/data/home/seondeok/.conda/envs/mmp/lib/python3.9/site-packages/mmengine/runner/runner.py", line 1735, in train
model = self.train_loop.run() # type: ignore
File "/data/home/seondeok/.conda/envs/mmp/lib/python3.9/site-packages/mmengine/runner/loops.py", line 96, in run
self.run_epoch()
File "/data/home/seondeok/.conda/envs/mmp/lib/python3.9/site-packages/mmengine/runner/loops.py", line 112, in run_epoch
self.run_iter(idx, data_batch)
File "/data/home/seondeok/.conda/envs/mmp/lib/python3.9/site-packages/mmengine/runner/loops.py", line 128, in run_iter
outputs = self.runner.model.train_step(
File "/data/home/seondeok/.conda/envs/mmp/lib/python3.9/site-packages/mmengine/model/base_model/base_model.py", line 113, in train_step
data = self.data_preprocessor(data, True)
File "/data/home/seondeok/.conda/envs/mmp/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/data/home/seondeok/.conda/envs/mmp/lib/python3.9/site-packages/mmengine/model/base_model/data_preprocessor.py", line 247, in forward
_batch_inputs = data['inputs']
KeyError: 'inputs'
Additional information
When I train topdown method models, it works well.
I also try to train bottom up method model in the same environment.
But error occur.
How to solve the problem?
I also add code of topdown heatmap model
This model train works well.
Prerequisite
Environment
mmcv 2.0.1 mmdet 3.1.0 mmengine 0.8.1 mmpose 1.1.0
Reproduces the problem - code sample
Reproduces the problem - command or script
python tools/train.py configs/body_2d_keypoint/associative_embedding/custom_model/ae_hrnet-w32_8xb24-300e_coco-512x512.py
Same file root configs/body_2d_keypoint/associative_embedding/coco/ae_hrnet-w32_8xb24-300e_coco-512x512.py
Reproduces the problem - error message
Additional information
When I train topdown method models, it works well. I also try to train bottom up method model in the same environment. But error occur. How to solve the problem?
I also add code of topdown heatmap model This model train works well.