STDC/configs/unet/fcn_unet_s5-d16_4x4_512x1024_160k_cityscapes.py
MengzhangLI 7f06e01d37
[Feature] Add UNet benchmark with multiple losses supervision (#1143)
* upload models and new configs

* fix hrf readme error

* fix hrf readme error

* add mDice of old models

* refactor configs
2021-12-23 21:38:51 +08:00

17 lines
420 B
Python

_base_ = [
'../_base_/models/fcn_unet_s5-d16.py', '../_base_/datasets/cityscapes.py',
'../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
]
model = dict(
decode_head=dict(num_classes=19),
auxiliary_head=dict(num_classes=19),
# model training and testing settings
train_cfg=dict(),
test_cfg=dict(mode='whole'))
data = dict(
samples_per_gpu=4,
workers_per_gpu=4,
)