Skip to content

Commit

Permalink
fix the bug "test_cfg specified in both outer field and model field "…
Browse files Browse the repository at this point in the history
… of pspnet config file;
  • Loading branch information
sennnnn committed Apr 16, 2021
1 parent 2de91d7 commit 9fb99b4
Show file tree
Hide file tree
Showing 4 changed files with 12 additions and 8 deletions.
5 changes: 3 additions & 2 deletions configs/pspnet/pspnet_r50-d8_480x480_40k_pascal_context.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
'../_base_/schedules/schedule_40k.py'
]
model = dict(
decode_head=dict(num_classes=60), auxiliary_head=dict(num_classes=60))
test_cfg = dict(mode='slide', crop_size=(480, 480), stride=(320, 320))
decode_head=dict(num_classes=60),
auxiliary_head=dict(num_classes=60),
test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
5 changes: 3 additions & 2 deletions configs/pspnet/pspnet_r50-d8_480x480_40k_pascal_context_59.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
'../_base_/schedules/schedule_40k.py'
]
model = dict(
decode_head=dict(num_classes=59), auxiliary_head=dict(num_classes=59))
test_cfg = dict(mode='slide', crop_size=(480, 480), stride=(320, 320))
decode_head=dict(num_classes=59),
auxiliary_head=dict(num_classes=59),
test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
5 changes: 3 additions & 2 deletions configs/pspnet/pspnet_r50-d8_480x480_80k_pascal_context.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
'../_base_/schedules/schedule_80k.py'
]
model = dict(
decode_head=dict(num_classes=60), auxiliary_head=dict(num_classes=60))
test_cfg = dict(mode='slide', crop_size=(480, 480), stride=(320, 320))
decode_head=dict(num_classes=60),
auxiliary_head=dict(num_classes=60),
test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
5 changes: 3 additions & 2 deletions configs/pspnet/pspnet_r50-d8_480x480_80k_pascal_context_59.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
'../_base_/schedules/schedule_80k.py'
]
model = dict(
decode_head=dict(num_classes=59), auxiliary_head=dict(num_classes=59))
test_cfg = dict(mode='slide', crop_size=(480, 480), stride=(320, 320))
decode_head=dict(num_classes=59),
auxiliary_head=dict(num_classes=59),
test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)

0 comments on commit 9fb99b4

Please sign in to comment.