Skip to content

Commit bfc8a3d

Browse files
committed
Fix wrong segmentation configs
1 parent f9841f6 commit bfc8a3d

6 files changed

Lines changed: 12 additions & 6 deletions

segmentation/configs/intern_vit_6b/few_shot/linear_intern_vit_6b_504_10k_ade20k_bs16_lr4e-5_1of8.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@
3333
qk_normalization=True,
3434
layerscale_force_fp32=False,
3535
freeze_vit=False,
36-
out_indices=[47]),
36+
out_indices=[47],
37+
pretrained=pretrained),
3738
decode_head=dict(
3839
_delete_=True,
3940
type='FCNHead',

segmentation/configs/intern_vit_6b/few_shot/linear_intern_vit_6b_504_20k_ade20k_bs16_lr4e-5_1of4.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@
3333
qk_normalization=True,
3434
layerscale_force_fp32=False,
3535
freeze_vit=False,
36-
out_indices=[47]),
36+
out_indices=[47],
37+
pretrained=pretrained),
3738
decode_head=dict(
3839
_delete_=True,
3940
type='FCNHead',

segmentation/configs/intern_vit_6b/few_shot/linear_intern_vit_6b_504_40k_ade20k_bs16_lr4e-5_1of2.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@
3333
qk_normalization=True,
3434
layerscale_force_fp32=False,
3535
freeze_vit=False,
36-
out_indices=[47]),
36+
out_indices=[47],
37+
pretrained=pretrained),
3738
decode_head=dict(
3839
_delete_=True,
3940
type='FCNHead',

segmentation/configs/intern_vit_6b/few_shot/linear_intern_vit_6b_504_5k_ade20k_bs16_lr4e-5_1of16.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@
3333
qk_normalization=True,
3434
layerscale_force_fp32=False,
3535
freeze_vit=False,
36-
out_indices=[47]),
36+
out_indices=[47],
37+
pretrained=pretrained),
3738
decode_head=dict(
3839
_delete_=True,
3940
type='FCNHead',

segmentation/configs/intern_vit_6b/few_shot/linear_intern_vit_6b_504_80k_ade20k_bs16_lr4e-5_1of1.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@
3333
qk_normalization=True,
3434
layerscale_force_fp32=False,
3535
freeze_vit=False,
36-
out_indices=[47]),
36+
out_indices=[47],
37+
pretrained=pretrained),
3738
decode_head=dict(
3839
_delete_=True,
3940
type='FCNHead',

segmentation/configs/intern_vit_6b/full_tuning/upernet_intern_vit_6b_504_80k_ade20k_bs16_lr4e-5.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,8 @@
3434
layerscale_force_fp32=False,
3535
with_fpn=True,
3636
freeze_vit=False,
37-
out_indices=[11, 23, 35, 47]),
37+
out_indices=[11, 23, 35, 47],
38+
pretrained=pretrained),
3839
decode_head=dict(num_classes=150,
3940
channels=1536,
4041
in_channels=[3200, 3200, 3200, 3200]),

0 commit comments

Comments
 (0)