Skip to content

Commit

Permalink
Change pretrained model link (#2133)
Browse files Browse the repository at this point in the history
  • Loading branch information
LareinaM authored and Tau-J committed Apr 6, 2023
1 parent dbe2854 commit 911c5df
Show file tree
Hide file tree
Showing 8 changed files with 16 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,8 @@
patch_cfg=dict(padding=2),
init_cfg=dict(
type='Pretrained',
checkpoint='pretrained/mae_pretrain_vit_base.pth'),
checkpoint='https://download.openmmlab.com/mmpose/'
'v1/pretrained_models/mae_pretrain_vit_base.pth'),
),
neck=dict(type='FeatureMapProcessor', scale_factor=4.0, apply_relu=True),
head=dict(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,8 @@
patch_cfg=dict(padding=2),
init_cfg=dict(
type='Pretrained',
checkpoint='pretrained/mae_pretrain_vit_base.pth'),
checkpoint='https://download.openmmlab.com/mmpose/'
'v1/pretrained_models/mae_pretrain_vit_base.pth'),
),
head=dict(
type='HeatmapHead',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,8 @@
patch_cfg=dict(padding=2),
init_cfg=dict(
type='Pretrained',
checkpoint='pretrained/mae_pretrain_vit_huge.pth'),
checkpoint='https://download.openmmlab.com/mmpose/'
'v1/pretrained_models/mae_pretrain_vit_huge.pth'),
),
neck=dict(type='FeatureMapProcessor', scale_factor=4.0, apply_relu=True),
head=dict(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,8 @@
patch_cfg=dict(padding=2),
init_cfg=dict(
type='Pretrained',
checkpoint='pretrained/mae_pretrain_vit_huge.pth'),
checkpoint='https://download.openmmlab.com/mmpose/'
'v1/pretrained_models/mae_pretrain_vit_huge.pth'),
),
head=dict(
type='HeatmapHead',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,8 @@
patch_cfg=dict(padding=2),
init_cfg=dict(
type='Pretrained',
checkpoint='pretrained/mae_pretrain_vit_large.pth'),
checkpoint='https://download.openmmlab.com/mmpose/'
'v1/pretrained_models/mae_pretrain_vit_large.pth'),
),
neck=dict(type='FeatureMapProcessor', scale_factor=4.0, apply_relu=True),
head=dict(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,8 @@
patch_cfg=dict(padding=2),
init_cfg=dict(
type='Pretrained',
checkpoint='pretrained/mae_pretrain_vit_large.pth'),
checkpoint='https://download.openmmlab.com/mmpose/'
'v1/pretrained_models/mae_pretrain_vit_large.pth'),
),
head=dict(
type='HeatmapHead',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,8 @@
patch_cfg=dict(padding=2),
init_cfg=dict(
type='Pretrained',
checkpoint='pretrained/mae_pretrain_vit_small.pth'),
checkpoint='https://download.openmmlab.com/mmpose/'
'v1/pretrained_models/mae_pretrain_vit_small.pth'),
),
neck=dict(type='FeatureMapProcessor', scale_factor=4.0, apply_relu=True),
head=dict(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,8 @@
patch_cfg=dict(padding=2),
init_cfg=dict(
type='Pretrained',
checkpoint='pretrained/mae_pretrain_vit_small.pth'),
checkpoint='https://download.openmmlab.com/mmpose/'
'v1/pretrained_models/mae_pretrain_vit_small.pth'),
),
head=dict(
type='HeatmapHead',
Expand Down

0 comments on commit 911c5df

Please sign in to comment.