diff --git a/configs/pp_liteseg/pp_liteseg_stdc1_cityscapes_1024x512_scale0.5_160k.yml b/configs/pp_liteseg/pp_liteseg_stdc1_cityscapes_1024x512_scale0.5_160k.yml index 2ee71e2d..943f21d8 100644 --- a/configs/pp_liteseg/pp_liteseg_stdc1_cityscapes_1024x512_scale0.5_160k.yml +++ b/configs/pp_liteseg/pp_liteseg_stdc1_cityscapes_1024x512_scale0.5_160k.yml @@ -13,14 +13,14 @@ lr_scheduler: loss: types: - - type: OhemCrossEntropyLoss + - type: CrossEntropyLoss weight: [0.3,0.5,0.2] min_kept: 130000 - type: OhemCrossEntropyLoss min_kept: 130000 - type: OhemCrossEntropyLoss min_kept: 130000 - coef: [1, 1, 1] + coef: [0.5, 1, 1] train_dataset: transforms: