Merge pull request #3946 from LDOUBLEV/dygraph

fix yml
This commit is contained in:
MissPenguin 2021-09-07 16:05:56 +08:00 committed by GitHub
commit e5dc3611cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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: