Image Segmentation
Transformers
PyTorch
upernet
Inference Endpoints
test2 / configs /_base_ /schedules /schedule_160k.py
mccaly's picture
Upload 660 files
b13b124
raw
history blame
382 Bytes
# optimizer
optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005)
optimizer_config = dict()
# learning policy
lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False)
# runtime settings
runner = dict(type='IterBasedRunner', max_iters=160000)
checkpoint_config = dict(by_epoch=False, interval=16000)
evaluation = dict(interval=16000, metric='mIoU')