Skip to content

Commit faaf29e

Browse files
authored
[Improvement] Move train_cfg/test_cfg inside model (open-mmlab#341)
* [Improvement] Move train_cfg/test_cfg inside model * fixed config dict * fixed config dict
1 parent 0ee3016 commit faaf29e

File tree

88 files changed

+266
-247
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

88 files changed

+266
-247
lines changed

configs/_base_/models/ann_r50-d8.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
norm_cfg=norm_cfg,
4141
align_corners=False,
4242
loss_decode=dict(
43-
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)))
44-
# model training and testing settings
45-
train_cfg = dict()
46-
test_cfg = dict(mode='whole')
43+
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
44+
# model training and testing settings
45+
train_cfg=dict(),
46+
test_cfg=dict(mode='whole'))

configs/_base_/models/apcnet_r50-d8.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
norm_cfg=norm_cfg,
3939
align_corners=False,
4040
loss_decode=dict(
41-
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)))
42-
# model training and testing settings
43-
train_cfg = dict()
44-
test_cfg = dict(mode='whole')
41+
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
42+
# model training and testing settings
43+
train_cfg=dict(),
44+
test_cfg=dict(mode='whole'))

configs/_base_/models/ccnet_r50-d8.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
norm_cfg=norm_cfg,
3939
align_corners=False,
4040
loss_decode=dict(
41-
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)))
42-
# model training and testing settings
43-
train_cfg = dict()
44-
test_cfg = dict(mode='whole')
41+
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
42+
# model training and testing settings
43+
train_cfg=dict(),
44+
test_cfg=dict(mode='whole'))

configs/_base_/models/cgnet.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
10.289121, 9.953208, 4.3097677, 9.490387, 7.674431, 9.396905,
3030
10.347791, 6.3927646, 10.226669, 10.241062, 10.280587,
3131
10.396974, 10.055647
32-
])))
33-
# model training and testing settings
34-
train_cfg = dict(sampler=None)
35-
test_cfg = dict(mode='whole')
32+
])),
33+
# model training and testing settings
34+
train_cfg=dict(sampler=None),
35+
test_cfg=dict(mode='whole'))

configs/_base_/models/danet_r50-d8.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
norm_cfg=norm_cfg,
3939
align_corners=False,
4040
loss_decode=dict(
41-
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)))
42-
# model training and testing settings
43-
train_cfg = dict()
44-
test_cfg = dict(mode='whole')
41+
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
42+
# model training and testing settings
43+
train_cfg=dict(),
44+
test_cfg=dict(mode='whole'))

configs/_base_/models/deeplabv3_r50-d8.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
norm_cfg=norm_cfg,
3939
align_corners=False,
4040
loss_decode=dict(
41-
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)))
42-
# model training and testing settings
43-
train_cfg = dict()
44-
test_cfg = dict(mode='whole')
41+
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
42+
# model training and testing settings
43+
train_cfg=dict(),
44+
test_cfg=dict(mode='whole'))

configs/_base_/models/deeplabv3_unet_s5-d16.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
norm_cfg=norm_cfg,
4545
align_corners=False,
4646
loss_decode=dict(
47-
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)))
48-
# model training and testing settings
49-
train_cfg = dict()
50-
test_cfg = dict(mode='slide', crop_size=256, stride=170)
47+
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
48+
# model training and testing settings
49+
train_cfg=dict(),
50+
test_cfg=dict(mode='slide', crop_size=256, stride=170))

configs/_base_/models/deeplabv3plus_r50-d8.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
norm_cfg=norm_cfg,
4141
align_corners=False,
4242
loss_decode=dict(
43-
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)))
44-
# model training and testing settings
45-
train_cfg = dict()
46-
test_cfg = dict(mode='whole')
43+
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
44+
# model training and testing settings
45+
train_cfg=dict(),
46+
test_cfg=dict(mode='whole'))

configs/_base_/models/dmnet_r50-d8.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
norm_cfg=norm_cfg,
3939
align_corners=False,
4040
loss_decode=dict(
41-
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)))
42-
# model training and testing settings
43-
train_cfg = dict()
44-
test_cfg = dict(mode='whole')
41+
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
42+
# model training and testing settings
43+
train_cfg=dict(),
44+
test_cfg=dict(mode='whole'))

configs/_base_/models/dnl_r50-d8.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
norm_cfg=norm_cfg,
4141
align_corners=False,
4242
loss_decode=dict(
43-
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)))
44-
# model training and testing settings
45-
train_cfg = dict()
46-
test_cfg = dict(mode='whole')
43+
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
44+
# model training and testing settings
45+
train_cfg=dict(),
46+
test_cfg=dict(mode='whole'))

0 commit comments

Comments
 (0)