Skip to content

GitLab

Unverified Commit 88dbd1ae authored by Zhe Chen's avatar Zhe Chen Committed by GitHub
Browse files

Update upernet_internimage_h_896_160k_ade20k.py

parent 1929ef42
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
segmentation/configs/ade20k/upernet_internimage_h_896_160k_ade20k.py
View file @ 88dbd1ae
...@@ -29,7 +29,7 @@ model = dict( ...@@ -29,7 +29,7 @@ model = dict(
center_feature_scale=True, # for InternImage-H/G center_feature_scale=True, # for InternImage-H/G
with_cp=False, with_cp=False,
out_indices=(0, 1, 2, 3), out_indices=(0, 1, 2, 3),
init_cfg=None), #dict(type='Pretrained', checkpoint=pretrained)), init_cfg=dict(type='Pretrained', checkpoint=pretrained)),
decode_head=dict(num_classes=150, in_channels=[320, 640, 1280, 2560]), decode_head=dict(num_classes=150, in_channels=[320, 640, 1280, 2560]),
auxiliary_head=dict(num_classes=150, in_channels=1280), auxiliary_head=dict(num_classes=150, in_channels=1280),
test_cfg=dict(mode='whole')) test_cfg=dict(mode='whole'))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment