mmsegmentation/configs/dnlnet/dnl_r50-d8_769x769_80k_cityscapes.py
Han Hu 8ff866d60c
Add "disentangled non-local (DNL) neural networks" [ECCV2020] (#37)
* Add DNLHead

* add configs

* add weight decay mult

* add norm back

* Update README.md

* matched inference performance

* Fixed shape

* sep conv_out

* no norm

* add norm back

* complete model zoo

* add tests

* Add test forward

* Add more test

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-09-07 16:22:00 +08:00

13 lines
460 B
Python

_base_ = [
'../_base_/models/dnl_r50-d8.py',
'../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
'../_base_/schedules/schedule_80k.py'
]
model = dict(
decode_head=dict(align_corners=True),
auxiliary_head=dict(align_corners=True))
test_cfg = dict(mode='slide', crop_size=(769, 769), stride=(513, 513))
optimizer = dict(
paramwise_cfg=dict(
custom_keys=dict(theta=dict(wd_mult=0.), phi=dict(wd_mult=0.))))