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

Luxus1001

3286人浏览 · 2022-10-25 10:18:00

Luxus1001 · 2022-10-25 10:18:00 发布

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

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

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

捡起python之：简单的任务系统

工作需要加之ai技术的应用再一次使python全面称王，博主大二时学习了python，现在需要捡一捡，今天的小项目是：简单的任务系统。

2048 AI社区

【AI】实战案例：用提示词生成微服务架构设计的思路

本文通过三个实战案例（电商订单、物流跟踪、在线教育）详细讲解如何用AI提示词生成微服务架构设计思路。每个案例包含需求分析、提示词设计、AI输出解析三部分，重点说明如何通过明确业务场景、核心功能、非功能需求等要素，让AI输出可落地的设计方案。案例展示了服务拆分、接口设计、中间件选型、性能优化等关键环节，并提供可直接复用的技术栈配置建议，帮助开发者快速构建符合业务需求的微服务架构。