Skip to content

Commit 9387455

Browse files
committed
[FIX] fixing bug in test_cfg for U-Net _base_ script
tuples were required as default input, but int was given
1 parent 1bb58de commit 9387455

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

configs/_base_/models/deeplabv3_unet_s5-d16.py

Lines changed: 1 addition & 1 deletion
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

Lines changed: 1 addition & 1 deletion
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

Lines changed: 1 addition & 1 deletion
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)