From 8666fccb262cbe0ec2f5304ba44cb60b91b6e97d Mon Sep 17 00:00:00 2001 From: dingizyuan Date: Mon, 1 Sep 2025 09:43:23 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9DETR3D=E7=9A=84=E8=AE=AD?= =?UTF-8?q?=E7=BB=83=E8=84=9A=E6=9C=AC=E9=85=8D=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- model_examples/DETR3D/README.md | 1 + model_examples/DETR3D/test/train_8p_full.sh | 2 +- model_examples/DETR3D/test/train_8p_performance.sh | 4 ++-- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/model_examples/DETR3D/README.md b/model_examples/DETR3D/README.md index b39d719c..410e219f 100644 --- a/model_examples/DETR3D/README.md +++ b/model_examples/DETR3D/README.md @@ -225,6 +225,7 @@ cd model_examples/DETR3D/detr3d 2024.12.30：首次发布 2025.1.13: 性能优化 +2025.9.1: 更改训练脚本配置 # FAQ diff --git a/model_examples/DETR3D/test/train_8p_full.sh b/model_examples/DETR3D/test/train_8p_full.sh index c1382fcd..2abb42c0 100644 --- a/model_examples/DETR3D/test/train_8p_full.sh +++ b/model_examples/DETR3D/test/train_8p_full.sh @@ -1,6 +1,6 @@ # 网络名称,同目录名称,需要模型审视修改 Network="DETR3D" -batch_size=1 +batch_size=3 world_size=8 py_config="projects/configs/detr3d/detr3d_res101_gridmask.py" diff --git a/model_examples/DETR3D/test/train_8p_performance.sh b/model_examples/DETR3D/test/train_8p_performance.sh index bced6cb5..a1a88c89 100644 --- a/model_examples/DETR3D/test/train_8p_performance.sh +++ b/model_examples/DETR3D/test/train_8p_performance.sh @@ -1,6 +1,6 @@ # 网络名称,同目录名称,需要模型审视修改 Network="DETR3D" -batch_size=1 +batch_size=3 world_size=8 py_config="projects/configs/detr3d/detr3d_res101_gridmask.py" @@ -41,7 +41,7 @@ echo "end_time=$(date -d @${end_time} "+%Y-%m-%d %H:%M:%S")" e2e_time=$(( $end_time - $start_time )) # 模型单epoch的step数量 -total_step=3517 +total_step=1173 # 单迭代训练时长 avg_time=$(echo "scale=3; $e2e_time / $total_step" | bc) -- Gitee