运行/torch/src/pytorch_models/Inference/Classification/mobilenetv3/classify.py 代码报错

AttributeError: module 'torch_mlu' has no attribute 'Input'

具体报错信息为

Traceback (most recent call last):

  File "/torch/src/pytorch_models/Inference/Classification/mobilenetv3/classify.py", line 402, in <module>

    test_cls_network(args)

  File "/torch/src/pytorch_models/Inference/Classification/mobilenetv3/classify.py", line 191, in test_cls_network

    torch_mlu.Input(

AttributeError: module 'torch_mlu' has no attribute 'Input'

Process finished with exit code 1