YOLO在服务器上训练,日志文件中不显示GFLOPs参数的问题

YOLO模型在服务器上训练,日志文件中不显示GFLOPs参数的问题

最近用服务器训练了YOLOv5模型,准备写论文找数据时才发现日志文件中没有GFLOPs参数,而在笔记本上训练时则会有GFLOPs参数,如下所示:
在这里插入图片描述
在这里插入图片描述找了好久才发现是因为服务器的环境中缺少thop函数库,直接
pip install thop安装就好了

你可能感兴趣的:(Python,目标检测,python,深度学习,开发语言)