PaddlePaddle / PaddleYOLO

🚀🚀🚀 YOLO series of PaddlePaddle implementation, PP-YOLOE+, RT-DETR, YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOv10, YOLOX, YOLOv5u, YOLOv7u, YOLOv6Lite, RTMDet and so on. 🚀🚀🚀
https://github.com/PaddlePaddle/PaddleYOLO
GNU General Public License v3.0
547 stars 133 forks source link

Aistudio paddlepaddle2.5.0训练yolov8报错 #165

Closed LDX17 closed 6 months ago

LDX17 commented 1 year ago

问题确认 Search before asking

请提出你的问题 Please ask your question

Aistudio paddlepaddle2.5.0训练yolov8报错如下: image image 环境:Aistudio paddlepaddle2.5.0 paddleyolo2.6 代码改动: 1)yolov8_s_500e_coco-my.yml文件 BASE: [ '../datasets/kitticoco_detection.yml', '../runtime.yml', 'base/optimizer_500e.yml', 'base/yolov8_cspdarknet.yml', 'base/yolov8_reader.yml', ] depth_mult: 0.33 width_mult: 0.50

save_dir: /home/aistudio/output print_flops: true print_params: true log_iter: 50 snapshot_epoch: 10 weights: /home/aistudio/output/yolov8_s_500e_coco-my/model_final pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov8_s_500e_coco.pdparams

epoch: 200

LearningRate: base_lr: 0.00125 schedulers:

TrainReader: batch_size: 16 # default 8 gpus, total bs = 128

mosaic_epoch: &mosaic_epoch 190 # last 10 epochs close mosaic, totally 500 epochs as default

2)kitticoco_detection.yml文件 image

运行命令: image

nemonameless commented 1 year ago

0iter打印出log信息了吗?建议再自查下数据集制作。也可同数据集去训yolov3或ppyoloe看看是否会有同样问题。 换用paddle2.4.2版本,就没有0D Tensor那些警告信息。

LDX17 commented 1 year ago

0iter没有打印出log信息。同数据集去训yolov3或ppyoloe正常。aistudio没有对应的paddle2.4.2版本。请问如何去排除这些错误?

nemonameless commented 1 year ago

关闭mosaic或syncbn试试。 https://github.com/PaddlePaddle/PaddleYOLO/blob/release/2.6/configs/yolov8/_base_/yolov8_cspdarknet.yml#L2 注释或者

mosaic_epoch: &mosaic_epoch -1

- MosaicPerspective: {mosaic_prob: 0.0, target_size: *input_size}

最新PaddleYOLO代码已经修复0D Tensor那些警告信息,paddle版本都可正常训了。

LDX17 commented 1 year ago

能训练了,但是训了10epoch的结果如下: image 我加了COCO预训练权重的 image

nemonameless commented 6 months ago

你好,加载预训练权重后去训,建议可以每两三个epoch就看下精度。谢谢。