KeyError: 'mlm_ids'

[yangmengmeng123]

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]

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