e38eae3894
* bisenetv1 on cocostuff164k * change config_names & delete redundant keys * pretrain should before lr. * remove redundancy in bisenetv1_r50-d32
14 lines
400 B
Python
14 lines
400 B
Python
_base_ = [
|
|
'../_base_/models/bisenetv1_r18-d32.py',
|
|
'../_base_/datasets/coco-stuff164k.py', '../_base_/default_runtime.py',
|
|
'../_base_/schedules/schedule_160k.py'
|
|
]
|
|
model = dict(
|
|
decode_head=dict(num_classes=171),
|
|
auxiliary_head=[
|
|
dict(num_classes=171),
|
|
dict(num_classes=171),
|
|
])
|
|
lr_config = dict(warmup='linear', warmup_iters=1000)
|
|
optimizer = dict(lr=0.005)
|