mmsegmentation/configs/unet/unet-s5-d16_fcn_4xb4-ce-1.0-dice-3.0-40k_hrf-256x256.py
谢昕辰 a3a144a361
[Refactor] Update config names (#1964)
* rename ann configs

* update ann yml

* update

* update

* update

* update

* update

* update ann readme

* update

* update deeplabv3

* update readme

* fix yml

* fix beit
2022-08-26 18:48:56 +08:00

7 lines
257 B
Python

_base_ = './unet-s5-d16_fcn_4xb4-40k_hrf-256x256.py'
model = dict(
decode_head=dict(loss_decode=[
dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0),
dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0)
]))