6.2recignition_by_transformer-》ocr_by_transformer.py运行报错,求帮助。。。

[Twsaiae]

D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\nn_reduction.py:44: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))

epoch 0
train...
D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\nn_reduction.py:44: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))
D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\nn_reduction.py:44: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))
D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\nn_reduction.py:44: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))
D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\nn_reduction.py:44: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))
Epoch Step: 1 Loss: 5.298872 Tokens per Sec: 69.338173
Traceback (most recent call last):
File "E:/pycharmproject/dive-into-cv-pytorch-master/dive-into-cv-pytorch-master/code/chapter06_transformer/6.2_recognition_by_transformer/ocr_by_transformer.py", line 341, in
train_mean_loss = run_epoch(train_loader, ocr_model, loss_compute, device)
File "E:/pycharmproject/dive-into-cv-pytorch-master/dive-into-cv-pytorch-master/code/chapter06_transformer/6.2_recognition_by_transformer/ocr_by_transformer.py", line 230, in run_epoch
for i, batch in enumerate(data_loader):
File "D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\utils\data\dataloader.py", line 435, in next
data = self._next_data()
File "D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\utils\data\dataloader.py", line 1085, in _next_data
return self._process_data(data)
File "D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\utils\data\dataloader.py", line 1111, in _process_data
data.reraise()
File "D:\anaconda\envs\transformer_ocr\lib\site-packages\torch_utils.py", line 428, in reraise
raise self.exc_type(msg)
KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\utils\data_utils\worker.py", line 198, in _worker_loop
data = fetcher.fetch(index)
File "D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\anaconda\envs\transformer_ocr\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "E:\pycharmproject\dive-into-cv-pytorch-master\dive-into-cv-pytorch-master\code\chapter06_transformer\6.2_recognition_by_transformer\ocr_by_transformer.py", line 109, in getitem
gt.append(self.lbl2id_map[lbl])
KeyError: 'Â'

进程已结束,退出代码1