Commit d2bace5a authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix yml

parent bb24d748
...@@ -23,9 +23,10 @@ Architecture: ...@@ -23,9 +23,10 @@ Architecture:
name: MobileNetV3 name: MobileNetV3
scale: 0.5 scale: 0.5
model_name: large model_name: large
disable_se: True
Neck: Neck:
name: DBFPN name: DBFPN
out_channels: 256 out_channels: 96
Head: Head:
name: DBHead name: DBHead
k: 50 k: 50
...@@ -34,8 +35,8 @@ Loss: ...@@ -34,8 +35,8 @@ Loss:
name: DBLoss name: DBLoss
balance_loss: true balance_loss: true
main_loss_type: DiceLoss main_loss_type: DiceLoss
alpha: 5 #5 alpha: 5
beta: 10 #10 beta: 10
ohem_ratio: 3 ohem_ratio: 3
Optimizer: Optimizer:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment