mirrors
/
SegFormer
mirror of https://github.com/NVlabs/SegFormer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update

pull/1/head
xieenze 2021-06-13 01:32:38 +08:00
parent e3d8945a36
commit 9de3d690b3
20 changed files with 21 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
local_configs/_base_/schedules/schedule_160k_8gpu_adamw.py → local_configs/_base_/schedules/schedule_160k_adamw.py
View File

@ -1,9 +1,9 @@
# optimizer
optimizer = dict(type='AdamW', lr=0.02/100, weight_decay=0.0001)
optimizer = dict(type='AdamW', lr=0.0002, weight_decay=0.0001)
optimizer_config = dict()
# learning policy
lr_config = dict(policy='poly', power=0.9, min_lr=0.0, by_epoch=False)
# runtime settings
runner = dict(type='IterBasedRunner', max_iters=160000)
checkpoint_config = dict(by_epoch=False, interval=4000)
evaluation = dict(interval=2000, metric='mIoU')
evaluation = dict(interval=4000, metric='mIoU')

9
local_configs/_base_/schedules/schedule_40k_8gpu_sgd.py
View File

@ -1,9 +0,0 @@
# optimizer
optimizer = dict(type='SGD', lr=0.02, 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=40000)
checkpoint_config = dict(by_epoch=False, interval=4000)
evaluation = dict(interval=4000, metric='mIoU')

4
local_configs/_base_/schedules/schedule_40k_8gpu_adamw.py → local_configs/_base_/schedules/schedule_40k_adamw.py
View File

@ -1,9 +1,9 @@
# optimizer
optimizer = dict(type='AdamW', lr=0.02/100, weight_decay=0.0001)
optimizer = dict(type='AdamW', lr=0.0002, weight_decay=0.0001)
optimizer_config = dict()
# learning policy
lr_config = dict(policy='poly', power=0.9, min_lr=0.0, by_epoch=False)
# runtime settings
runner = dict(type='IterBasedRunner', max_iters=40000)
checkpoint_config = dict(by_epoch=False, interval=4000)
evaluation = dict(interval=2000, metric='mIoU')
evaluation = dict(interval=4000, metric='mIoU')

9
local_configs/_base_/schedules/schedule_80k_8gpu_sgd.py
View File

@ -1,9 +0,0 @@
# optimizer
optimizer = dict(type='SGD', lr=0.02, 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=80000)
checkpoint_config = dict(by_epoch=False, interval=8000)
evaluation = dict(interval=8000, metric='mIoU')

4
local_configs/_base_/schedules/schedule_80k_8gpu_adamw.py → local_configs/_base_/schedules/schedule_80k_adamw.py
View File

@ -1,9 +1,9 @@
# optimizer
optimizer = dict(type='AdamW', lr=0.02/100, weight_decay=0.0001)
optimizer = dict(type='AdamW', lr=0.0002, weight_decay=0.0001)
optimizer_config = dict()
# learning policy
lr_config = dict(policy='poly', power=0.9, min_lr=0.0, by_epoch=False)
# runtime settings
runner = dict(type='IterBasedRunner', max_iters=80000)
checkpoint_config = dict(by_epoch=False, interval=4000)
evaluation = dict(interval=2000, metric='mIoU')
evaluation = dict(interval=4000, metric='mIoU')

3
local_configs/segformer/B0/segformer.b0.1024x1024.city.160k.py
View File

@ -1,9 +1,8 @@
#新开始一个
_base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/cityscapes_1024x1024_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B0/segformer.b0.512x1024.city.160k.py
View File

@ -1,7 +1,7 @@
_base_ = [
'../../_base_/models/segformer.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B0/segformer.b0.512x512.ade.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/ade20k_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B0/segformer.b0.640x1280.city.160k.py
View File

@ -1,7 +1,7 @@
_base_ = [
'../../_base_/models/segformer.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B0/segformer.b0.768x768.city.160k.py
View File

@ -1,7 +1,7 @@
_base_ = [
'../../_base_/models/segformer.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B1/segformer.b1.1024x1024.city.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/cityscapes_1024x1024_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B1/segformer.b1.512x512.ade.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/ade20k_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B2/segformer.b2.1024x1024.city.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/cityscapes_1024x1024_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B2/segformer.b2.512x512.ade.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/ade20k_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B3/segformer.b3.1024x1024.city.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/cityscapes_1024x1024_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B3/segformer.b3.512x512.ade.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/ade20k_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B4/segformer.b4.1024x1024.city.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/cityscapes_1024x1024_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B4/segformer.b4.512x512.ade.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/ade20k_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B5/segformer.b5.1024x1024.city.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
'../../_base_/datasets/cityscapes_1024x1024_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# model settings

2
local_configs/segformer/B5/segformer.b5.640x640.ade.160k.py
View File

@ -2,7 +2,7 @@ _base_ = [
'../../_base_/models/segformer.py',
# '../../_base_/datasets/ade20k_repeat.py',
'../../_base_/default_runtime.py',
'../../_base_/schedules/schedule_160k_8gpu_adamw.py'
'../../_base_/schedules/schedule_160k_adamw.py'
]
# data settings