Skip to content

Commit bd87c92

Browse files
authored
Merge pull request #1 from smttsp/ISSUE-2337/bug/unet_param
[FIX] fixing bug in `test_cfg` for U-Net _base_ script
2 parents 1bb58de + 9387455 commit bd87c92

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

configs/_base_/models/deeplabv3_unet_s5-d16.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -47,4 +47,4 @@
4747
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
4848
# model training and testing settings
4949
train_cfg=dict(),
50-
test_cfg=dict(mode='slide', crop_size=256, stride=170))
50+
test_cfg=dict(mode='slide', crop_size=(256, 256), stride=(170, 170)))

configs/_base_/models/fcn_unet_s5-d16.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -48,4 +48,4 @@
4848
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
4949
# model training and testing settings
5050
train_cfg=dict(),
51-
test_cfg=dict(mode='slide', crop_size=256, stride=170))
51+
test_cfg=dict(mode='slide', crop_size=(256, 256), stride=(170, 170)))

configs/_base_/models/pspnet_unet_s5-d16.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -47,4 +47,4 @@
4747
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
4848
# model training and testing settings
4949
train_cfg=dict(),
50-
test_cfg=dict(mode='slide', crop_size=256, stride=170))
50+
test_cfg=dict(mode='slide', crop_size=(256, 256), stride=(170, 170)))

0 commit comments

Comments
 (0)