|
|
_base_ = [ |
|
|
'/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/bit_r18.py', |
|
|
'/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_medium_512x512_100k_cgwx.py'] |
|
|
|
|
|
param_scheduler = [ |
|
|
dict( |
|
|
type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000), |
|
|
dict( |
|
|
type='PolyLR', |
|
|
power=1.0, |
|
|
begin=1000, |
|
|
end=100000, |
|
|
eta_min=0.0, |
|
|
by_epoch=False, |
|
|
) |
|
|
] |
|
|
|
|
|
train_cfg = dict(type='IterBasedTrainLoop', max_iters=100000, val_interval=1000) |
|
|
val_cfg = dict(type='ValLoop') |
|
|
test_cfg = dict(type='TestLoop') |
|
|
default_hooks = dict( |
|
|
timer=dict(type='IterTimerHook'), |
|
|
logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False), |
|
|
param_scheduler=dict(type='ParamSchedulerHook'), |
|
|
checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000, |
|
|
save_best='mIoU'), |
|
|
sampler_seed=dict(type='DistSamplerSeedHook'), |
|
|
visualization=dict(type='CDVisualizationHook', interval=1, |
|
|
img_shape=(512, 512, 3))) |
