From 244d0bce0246a51c77ce9710e4b4fe75b0b1cd82 Mon Sep 17 00:00:00 2001 From: tackhwa <55059307+tackhwa@users.noreply.github.com> Date: Sun, 19 May 2024 23:04:55 +0800 Subject: [PATCH 1/5] fix old MMCV config file format for mobilenetv3 --- ...v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py | 9 ++++++++- ...bilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py | 8 +++++++- 2 files changed, 15 insertions(+), 2 deletions(-) diff --git a/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py b/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py index 8dcbc3395f..5723fdcba9 100644 --- a/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py +++ b/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py @@ -4,10 +4,17 @@ ] crop_size = (512, 1024) data_preprocessor = dict(size=crop_size) + +#optimizer +optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001) +optim_wrapper = dict(type='OptimWrapper', optimizer=optimizer, clip_grad=None) + # Re-config the data sampler. model = dict(data_preprocessor=data_preprocessor) 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) +train_cfg = dict(type="IterBasedTrainLoop", max_iters=320000, val_interval=32000) +val_cfg = dict(type='ValLoop') +test_cfg = dict(type='TestLoop') diff --git a/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py b/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py index cd84265f32..229f22e822 100644 --- a/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py +++ b/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py @@ -8,9 +8,15 @@ data_preprocessor=data_preprocessor, pretrained='open-mmlab://contrib/mobilenet_v3_large') +#optimizer +optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001) +optim_wrapper = dict(type='OptimWrapper', optimizer=optimizer, clip_grad=None) + # 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) +train_cfg = dict(type="IterBasedTrainLoop", max_iters=320000, val_interval=32000) +val_cfg = dict(type='ValLoop') +test_cfg = dict(type='TestLoop') From 485f7c5d97f0b25b60b3137bca1a8d9b27d4f318 Mon Sep 17 00:00:00 2001 From: tackhwa <55059307+tackhwa@users.noreply.github.com> Date: Mon, 20 May 2024 02:12:00 +0800 Subject: [PATCH 2/5] fix old MMCV config file format for mobilenetv3 --- .pre-commit-config.yaml | 2 +- ...net-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py | 5 +++-- .../mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py | 5 +++-- 3 files changed, 7 insertions(+), 5 deletions(-) diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 337e90cdca..4124f6313d 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -64,4 +64,4 @@ repos: hooks: - id: check-algo-readme - id: check-copyright - args: ["mmseg", "tools", "tests", "demo"] # the dir_to_check with expected directory to check + args: ["mmseg", "tools", "tests", "demo"] # the dir_to_check with expected directory to check \ No newline at end of file diff --git a/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py b/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py index 5723fdcba9..d0a637b0e1 100644 --- a/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py +++ b/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py @@ -5,7 +5,7 @@ crop_size = (512, 1024) data_preprocessor = dict(size=crop_size) -#optimizer +# optimizer optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001) optim_wrapper = dict(type='OptimWrapper', optimizer=optimizer, clip_grad=None) @@ -15,6 +15,7 @@ val_dataloader = dict(batch_size=1, num_workers=4) test_dataloader = val_dataloader -train_cfg = dict(type="IterBasedTrainLoop", max_iters=320000, val_interval=32000) +train_cfg = dict( + type='IterBasedTrainLoop', max_iters=320000, val_interval=32000) val_cfg = dict(type='ValLoop') test_cfg = dict(type='TestLoop') diff --git a/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py b/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py index 229f22e822..f92f2b3412 100644 --- a/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py +++ b/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py @@ -8,7 +8,7 @@ data_preprocessor=data_preprocessor, pretrained='open-mmlab://contrib/mobilenet_v3_large') -#optimizer +# optimizer optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001) optim_wrapper = dict(type='OptimWrapper', optimizer=optimizer, clip_grad=None) @@ -17,6 +17,7 @@ val_dataloader = dict(batch_size=1, num_workers=4) test_dataloader = val_dataloader -train_cfg = dict(type="IterBasedTrainLoop", max_iters=320000, val_interval=32000) +train_cfg = dict( + type='IterBasedTrainLoop', max_iters=320000, val_interval=32000) val_cfg = dict(type='ValLoop') test_cfg = dict(type='TestLoop') From 812401ad96d2ff2f813325fee453b6b9d8f27129 Mon Sep 17 00:00:00 2001 From: tackhwa <55059307+tackhwa@users.noreply.github.com> Date: Mon, 20 May 2024 17:41:21 +0800 Subject: [PATCH 3/5] Update .pre-commit-config.yaml --- .pre-commit-config.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 4124f6313d..337e90cdca 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -64,4 +64,4 @@ repos: hooks: - id: check-algo-readme - id: check-copyright - args: ["mmseg", "tools", "tests", "demo"] # the dir_to_check with expected directory to check \ No newline at end of file + args: ["mmseg", "tools", "tests", "demo"] # the dir_to_check with expected directory to check From c6e3b23ab5a204254fa851936452907ab056df3b Mon Sep 17 00:00:00 2001 From: tackhwa <55059307+tackhwa@users.noreply.github.com> Date: Mon, 20 May 2024 18:03:53 +0800 Subject: [PATCH 4/5] Update mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py --- ...8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py b/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py index d0a637b0e1..766e3551a7 100644 --- a/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py +++ b/configs/mobilenet_v3/mobilenet-v3-d8-scratch_lraspp_4xb4-320k_cityscapes-512x1024.py @@ -1,21 +1,11 @@ _base_ = [ '../_base_/models/lraspp_m-v3-d8.py', '../_base_/datasets/cityscapes.py', - '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' + '../_base_/default_runtime.py', '../_base_/schedules/schedule_320k.py' ] crop_size = (512, 1024) data_preprocessor = dict(size=crop_size) - -# optimizer -optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001) -optim_wrapper = dict(type='OptimWrapper', optimizer=optimizer, clip_grad=None) - # Re-config the data sampler. model = dict(data_preprocessor=data_preprocessor) train_dataloader = dict(batch_size=4, num_workers=4) val_dataloader = dict(batch_size=1, num_workers=4) test_dataloader = val_dataloader - -train_cfg = dict( - type='IterBasedTrainLoop', max_iters=320000, val_interval=32000) -val_cfg = dict(type='ValLoop') -test_cfg = dict(type='TestLoop') From 43d8752f8d55b830a899b75b8de8e725141c3bad Mon Sep 17 00:00:00 2001 From: tackhwa <55059307+tackhwa@users.noreply.github.com> Date: Mon, 20 May 2024 18:08:20 +0800 Subject: [PATCH 5/5] Update mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py --- ...enet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) diff --git a/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py b/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py index f92f2b3412..25beee953d 100644 --- a/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py +++ b/configs/mobilenet_v3/mobilenet-v3-d8_lraspp_4xb4-320k_cityscapes-512x1024.py @@ -1,6 +1,6 @@ _base_ = [ '../_base_/models/lraspp_m-v3-d8.py', '../_base_/datasets/cityscapes.py', - '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' + '../_base_/default_runtime.py', '../_base_/schedules/schedule_320k.py' ] crop_size = (512, 1024) data_preprocessor = dict(size=crop_size) @@ -8,16 +8,7 @@ data_preprocessor=data_preprocessor, pretrained='open-mmlab://contrib/mobilenet_v3_large') -# optimizer -optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001) -optim_wrapper = dict(type='OptimWrapper', optimizer=optimizer, clip_grad=None) - # 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 - -train_cfg = dict( - type='IterBasedTrainLoop', max_iters=320000, val_interval=32000) -val_cfg = dict(type='ValLoop') -test_cfg = dict(type='TestLoop')