| _base_ = [ | |
| '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/cgnet.py', | |
| '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/standard_512x512_200k_cgwx.py'] | |
| # optimizer | |
| optimizer = dict( | |
| type='AdamW', | |
| lr=5e-4, | |
| betas=(0.9, 0.999), | |
| weight_decay=0.0025) | |
| optim_wrapper = dict( | |
| _delete_=True, | |
| type='OptimWrapper', | |
| optimizer=optimizer) | |
| crop_size = (512, 512) | |
| model = dict( | |
| test_cfg=dict(mode='slide', crop_size=crop_size, stride=(crop_size[0]//2, crop_size[1]//2)), | |
| ) | |