pull/3946/head
LDOUBLEV 2021-09-07 08:00:23 +00:00
parent bb24d74826
commit d2bace5a0e
1 changed files with 4 additions and 3 deletions

View File

@ -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: