Skip to content

【25-Q4-生态建设】模型迁移-研发效能部-模型训练-在PyTorch框架上支持 Fast R-CNN 在 COCO 上的训练#458

Open
x0212wwl wants to merge 3 commits intoTecorigin:mainfrom
x0212wwl:fast_rcnn
Open

【25-Q4-生态建设】模型迁移-研发效能部-模型训练-在PyTorch框架上支持 Fast R-CNN 在 COCO 上的训练#458
x0212wwl wants to merge 3 commits intoTecorigin:mainfrom
x0212wwl:fast_rcnn

Conversation

@x0212wwl
Copy link

@x0212wwl x0212wwl commented Jan 4, 2026

● 当前软件栈版本:
image

● 源码参考链接:https://github.com/open-mmlab/mmdetection/blob/main/README_zh-CN.md
● commit id:x0212wwl@ https://github.com/x0212wwl
● 工作目录:PyTorch/build-in/Detection/fast_rcnn/
● 训练内容:使用1张TECO_AICARD_01芯片,在PyTorch框架上支持 RPN 在 COCO 数据集上的训练。
● 运行脚本如下:
SDAA_VISIBLE_DEVICES=8,9,10,11 python weloTrain.py --name train --model fastrcnn --steps 100 --datapath ../data --batch-size 16 | tee fastrcnnCOCOSdaa.log

● 100iters损失:
image
MeanRelativeError: 0.06133527711382374
MeanAbsoluteError: -0.011225999999999998
Rule,mean_absolute_error -0.011225999999999998
pass mean_relative_error=0.06133527711382374 <= 0.05 or mean_absolute_error=-0.011225999999999998 <= 0.0002

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant