Comments (6)
改动:只改了配置文件
coco_detection.yml 文件中 num_classes: 4
yolov7p6_elannet.yml 文件中 YOLOv7Head: anchors: [[29, 35], [41, 62], [79, 37], [96, 101], [185, 58], [155, 174], [298, 125], [267, 290], [621, 176], [414, 402], [880, 414], [945, 949]] (tools/anchor_cluster.py 生成)
yolov7p6_reader.yml 文件中 worker_num: 1 ; TrainReader:下batch_size: 4 ; EvalReader:下 batch_size: 4
版本:
paddlepaddle-gpu 2.3.2
paddledet 2.4.0
paddle-bfloat 0.1.7
python 3.6.2
cudatoolkit 11.2.2
cudnn 8.1.0.77
服务器环境:
Tesla P40 服务器 Linux gpu 4.15.0-142-generic系统
Anaconda下使用conda创建的python3.6的虚拟环境,然后安装了paddle2.3.2 和paddledetection_yoloseries
训练命令:export CUDA_VISIBLE_DEVICES=7
python --log_dir=./yolo7_dygraph/ tools/train.py -c configs/yolov7/yolov7p6_e6e_300e_coco.yml --eval -o use_gpu=true
from paddleyolo.
上面训练命令采用多卡会造成另一个错误,也已经在问题区进行了提问,期待答复
from paddleyolo.
from paddleyolo.
我看到gitee上也有人提出这个问题,但是没检索到解决方法,我的情况和他的类似!
from paddleyolo.
提问请遵循issue模板,说明下改动了什么,版本和环境是什么,以便排查问题。
from paddleyolo.
一开始训就显存一直上升直到崩溃,就说明bs过大了,调小点就行。
训练命令加上--amp
混合精度训练,加上可以同bs下减小显存占用。
yolov7p6_e6e是最大的模型,不是一般显卡可以训起来的,训自己的数据集注意要加载coco权重做预训练不要从头训,首页readme写明了。
from paddleyolo.
Related Issues (20)
- ins_seg分支可用吗 HOT 1
- yolov5u infer出错 HOT 1
- 多卡训练卡住不继续运行 HOT 1
- PaddleYOLOv8是否支持关键点检测 HOT 2
- YOLOv8在客制化数据集上训练出现SystemError HOT 3
- 是否支持yolov8的实例分割训练和检测 HOT 1
- 目标检测:忽略某些模糊的目标 HOT 3
- yolov8训练好的模型如何在vc++生成的项目中调用? HOT 3
- 使用预训练的模型进行推理,有识别结果。基于预训练的模型训练Visdrone数据集后,使用训练后的模型进行推理,检测多张该数据集的图片,均无识别结果。 HOT 5
- 单机多卡,拉去gpu版本的docker,进行版本验证时只能使用单张gpu,已安装nccl 2.8.3 HOT 3
- 现在支持训练yolov8了吗? HOT 3
- Single-class yolov7 training hangs HOT 2
- ins_seg分支YOLOv5、v8和PPYOLOE seg目前支持训练了吗 HOT 3
- PaddleYolo训练时报错 HOT 4
- ppyoloe的8卡长训结果和官方对不齐 HOT 1
- yolov5无法复现PaddleYOLO官方结果 HOT 4
- 利好Paddle......yolov5,yolov8表示,只要你用了模型,包含ONNX模型,都必须要开源 HOT 1
- 模型保存时报错 HOT 3
- Export model for inference HOT 1
- ppyoloe_seg的实例分割模型下载链接失效了 HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from paddleyolo.