File size: 504 Bytes
b1b30c0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
_base_ = [
'/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/lightcdnet.py',
'/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/standard_512x512_200k_cgwx.py']
model = dict(
decode_head=dict(
sampler=dict(type='mmseg.OHEMPixelSampler', thresh=0.7, min_kept=100000)))
# optimizer
optimizer = dict(
type='AdamW',
lr=0.003,
betas=(0.9, 0.999),
weight_decay=0.05)
optim_wrapper = dict(
_delete_=True,
type='OptimWrapper',
optimizer=optimizer)
|