From d2bace5a0e068909bb8b65a1df7a4801c9ea7645 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 7 Sep 2021 08:00:23 +0000 Subject: [PATCH] fix yml --- tests/configs/det_mv3_db.yml | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/configs/det_mv3_db.yml b/tests/configs/det_mv3_db.yml index 08456871..d6d4c26c 100644 --- a/tests/configs/det_mv3_db.yml +++ b/tests/configs/det_mv3_db.yml @@ -23,9 +23,10 @@ Architecture: name: MobileNetV3 scale: 0.5 model_name: large + disable_se: True Neck: name: DBFPN - out_channels: 256 + out_channels: 96 Head: name: DBHead k: 50 @@ -34,8 +35,8 @@ Loss: name: DBLoss balance_loss: true main_loss_type: DiceLoss - alpha: 5 #5 - beta: 10 #10 + alpha: 5 + beta: 10 ohem_ratio: 3 Optimizer: