KIE Training error

[riyajatar93]

i am trying to train kie but i am getting

Traceback (most recent call last):
File "tools/train.py", line 227, in
main(config, device, logger, vdl_writer)
File "tools/train.py", line 198, in main
program.train(config, train_dataloader, valid_dataloader, device, model,
File "/home/riyaj-atar/Desktop/PaddleOCR/tools/program.py", line 308, in train
loss = loss_class(preds, batch)
File "/home/riyaj-atar/anaconda3/envs/ocr_env/lib/python3.8/site-packages/paddle/nn/layer/layers.py", line 1254, in call
return self.forward(*inputs, **kwargs)
File "/home/riyaj-atar/Desktop/PaddleOCR/ppocr/losses/kie_sdmgr_loss.py", line 110, in forward
acc_node=self.accuracy(
File "/home/riyaj-atar/Desktop/PaddleOCR/ppocr/losses/kie_sdmgr_loss.py", line 73, in accuracy
accu = [pred.new_tensor(0.) for i in range(len(topk))]
File "/home/riyaj-atar/Desktop/PaddleOCR/ppocr/losses/kie_sdmgr_loss.py", line 73, in
accu = [pred.new_tensor(0.) for i in range(len(topk))]
AttributeError: 'Tensor' object has no attribute 'new_tensor'