Spaces:
Runtime error
Runtime error
snnetv2-semantic-segmentation
/
configs
/mobilenet_v3
/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py
_base_ = [ | |
'../_base_/models/lraspp_m-v3-d8.py', '../_base_/datasets/cityscapes.py', | |
'../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' | |
] | |
crop_size = (512, 1024) | |
data_preprocessor = dict(size=crop_size) | |
model = dict( | |
data_preprocessor=data_preprocessor, | |
pretrained='open-mmlab://contrib/mobilenet_v3_large') | |
# Re-config the data sampler. | |
train_dataloader = dict(batch_size=4, num_workers=4) | |
val_dataloader = dict(batch_size=1, num_workers=4) | |
test_dataloader = val_dataloader | |
runner = dict(type='IterBasedRunner', max_iters=320000) | |