Image Segmentation
Transformers
PyTorch
upernet
Inference Endpoints
test2 / configs /_base_ /schedules /schedule_20k.py
mccaly's picture
Upload 660 files
b13b124
raw
history blame
379 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=20000)
checkpoint_config = dict(by_epoch=False, interval=2000)
evaluation = dict(interval=2000, metric='mIoU')