UnboundLocalError: local variable 'idxs' referenced before assignment

[DDxk369]

Hello, when I run bash ./run_code.sh, the following error has occurred, and how can I solve this problem?

Traceback (most recent call last): File "main.py", line 497, in main(args) File "main.py", line 448, in main args=args File "/content/evit/engine.py", line 56, in train_one_epoch outputs = model(samples, keep_rate) File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 1110, in _call_impl return forward_call(*input, **kwargs) File "/usr/local/lib/python3.7/dist-packages/torch/nn/parallel/distributed.py", line 963, in forward output = self.module(*inputs[0], **kwargs[0]) File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 1110, in _call_impl return forward_call(*input, **kwargs) File "/content/evit/evit.py", line 442, in forward x, _, idxs = self.forward_features(x, keep_rate, tokens, get_idx) File "/content/evit/evit.py", line 437, in forward_features return self.pre_logits(x[:, 0]), left_tokens, idxs UnboundLocalError: local variable 'idxs' referenced before assignment