pycharm中用yolov5跑训练时AttributeError报错

问题遇到的现象和发生背景

我在pycharm中用yolov5跑训练时报错了，请问该怎么弄呀？

用代码块功能插入代码，请勿粘贴截图

AttributeError: Can't get attribute 'ClassificationModel' on \\Users\\86158\\Desktop\\yolov5-5.0\\models\\yolo.py'>

大概率是路径问题，导致定义ClassificationModel的模块没导入