mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Update metaformers.py

Browse Source
This commit is contained in:
Fredo Guan 2023-01-28 15:09:12 -08:00
parent 729533e966
commit 7c04f6dc75
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
timm/models/metaformers.py
View File

@ -543,7 +543,7 @@ class MetaFormer(nn.Module):
dp_rates=[x.item() for x in torch.linspace(0, drop_path_rate, sum(depths))]
self.patch_embed = Downsampling(
self.stem = Downsampling(
in_chans,
dims[0],
kernel_size=7,
@ -670,6 +670,7 @@ def checkpoint_filter_fn(state_dict, model):
k = k.replace('patch_embed.proj', 'patch_embed.conv')
k = re.sub(r'([0-9]+).([0-9]+)', r'\1.blocks.\2', k)
k = k.replace('stages.0.downsample', 'patch_embed')
k = k.replace('patch_embed', 'stem')
k = re.sub(r'^head', 'head.fc', k)
k = re.sub(r'^norm', 'head.norm', k)
out_dict[k] = v