Giter VIP home page Giter VIP logo

Comments (6)

ZhangLe-fighting avatar ZhangLe-fighting commented on May 14, 2024 2

改动:只改了配置文件
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.

ZhangLe-fighting avatar ZhangLe-fighting commented on May 14, 2024 1

上面训练命令采用多卡会造成另一个错误,也已经在问题区进行了提问,期待答复

from paddleyolo.

ZhangLe-fighting avatar ZhangLe-fighting commented on May 14, 2024

Snipaste_2022-09-05_18-09-41

from paddleyolo.

ZhangLe-fighting avatar ZhangLe-fighting commented on May 14, 2024

我看到gitee上也有人提出这个问题,但是没检索到解决方法,我的情况和他的类似!

from paddleyolo.

nemonameless avatar nemonameless commented on May 14, 2024

提问请遵循issue模板,说明下改动了什么,版本和环境是什么,以便排查问题。

from paddleyolo.

nemonameless avatar nemonameless commented on May 14, 2024

一开始训就显存一直上升直到崩溃,就说明bs过大了,调小点就行。
训练命令加上--amp混合精度训练,加上可以同bs下减小显存占用。
yolov7p6_e6e是最大的模型,不是一般显卡可以训起来的,训自己的数据集注意要加载coco权重做预训练不要从头训,首页readme写明了。

from paddleyolo.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo 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.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.