关于平台上yolov5那个项目中train_v6.py

---

请问还是像原版yolov5那种命令运行吗，我试了一下训练自己的数据集，发现有错误，请问怎么解决？

对应着spinenet_yolo.py中
我明明装了pytorch1.7但是这程序里显示无法导入是怎么回事

请问谁知道yolov6 剪枝后的模型在哪下载呢？

@carlsummer 谢谢，安装了包之后就解决了

pip install nbnb