From e9cfb1876e5dabfd116526639911905cc8e7a6fd Mon Sep 17 00:00:00 2001 From: shangliang Xu Date: Thu, 3 Mar 2022 18:36:04 +0800 Subject: [PATCH] [DETR] fix find_unused_parameters in yaml (#5291) --- configs/deformable_detr/README.md | 2 +- configs/deformable_detr/deformable_detr_r50_1x_coco.yml | 1 + configs/detr/README.md | 2 +- configs/detr/detr_r50_1x_coco.yml | 1 + 4 files changed, 4 insertions(+), 2 deletions(-) diff --git a/configs/deformable_detr/README.md b/configs/deformable_detr/README.md index 41ae056e3..ea6840fb5 100644 --- a/configs/deformable_detr/README.md +++ b/configs/deformable_detr/README.md @@ -20,7 +20,7 @@ Deformable DETR is an object detection model based on DETR. We reproduced the mo GPU multi-card training ```bash export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 -python -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/deformable_detr/deformable_detr_r50_1x_coco.yml --fleet -o find_unused_parameters=True +python -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/deformable_detr/deformable_detr_r50_1x_coco.yml --fleet ``` ## Citations diff --git a/configs/deformable_detr/deformable_detr_r50_1x_coco.yml b/configs/deformable_detr/deformable_detr_r50_1x_coco.yml index f924c4e2e..4ca749106 100644 --- a/configs/deformable_detr/deformable_detr_r50_1x_coco.yml +++ b/configs/deformable_detr/deformable_detr_r50_1x_coco.yml @@ -6,3 +6,4 @@ _BASE_: [ '_base_/deformable_detr_reader.yml', ] weights: output/deformable_detr_r50_1x_coco/model_final +find_unused_parameters: True diff --git a/configs/detr/README.md b/configs/detr/README.md index cc0470e65..2dc370f3a 100644 --- a/configs/detr/README.md +++ b/configs/detr/README.md @@ -20,7 +20,7 @@ DETR is an object detection model based on transformer. We reproduced the model GPU multi-card training ```bash export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 -python -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/detr/detr_r50_1x_coco.yml --fleet -o find_unused_parameters=True +python -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/detr/detr_r50_1x_coco.yml --fleet ``` ## Citations diff --git a/configs/detr/detr_r50_1x_coco.yml b/configs/detr/detr_r50_1x_coco.yml index 0df91bb56..d8838fac0 100644 --- a/configs/detr/detr_r50_1x_coco.yml +++ b/configs/detr/detr_r50_1x_coco.yml @@ -6,3 +6,4 @@ _BASE_: [ '_base_/detr_reader.yml', ] weights: output/detr_r50_1x_coco/model_final +find_unused_parameters: True -- GitLab