search

anosorae / IRRA

Cross-Modal Implicit Relation Reasoning and Aligning for Text-to-Image Person Retrieval (CVPR 2023)
MIT License
194 stars 27 forks source link

KeyError: 'mlm_ids' #27

Closed yangmengmeng123 closed 11 months ago

yangmengmeng123 commented 11 months ago

When I run train.py, the following error appears:

Traceback (most recent call last): File "f:\Workspace\0research0\IRRA-main\train.py", line 77, in do_train(start_epoch, args, model, train_loader, evaluator, optimizer, scheduler, checkpointer)
File "f:\Workspace\0research0\IRRA-main\processor\processor.py", line 50, in do_train ret = model(batch) ^^^^^^^^^^^^ File "F:\env\pytorch\Lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, *kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "F:\env\pytorch\Lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl return forward_call(args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "f:\Workspace\0research0\IRRA-main\model\build.py", line 126, in forward mlm_ids = batch['mlm_ids']


KeyError: 'mlm_ids'
hanjx520 commented 1 month ago

When I run train.py, the following error appears:

Traceback (most recent call last): File "f:\Workspace\0research0\IRRA-main\train.py", line 77, in do_train(start_epoch, args, model, train_loader, evaluator, optimizer, scheduler, checkpointer) File "f:\Workspace\0research0\IRRA-main\processor\processor.py", line 50, in do_train ret = model(batch) ^^^^^^^^^^^^ File "F:\env\pytorch\Lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, *kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "F:\env\pytorch\Lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl return forward_call(args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "f:\Workspace\0research0\IRRA-main\model\build.py", line 126, in forward mlm_ids = batch['mlm_ids'] ~^^^^^^^^^^^ KeyError: 'mlm_ids'

Has this problem been solved now