Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update metaformers.py

Browse Source
pull/1647/head
Fredo Guan 1 year ago
parent e2a9408dd0
commit a73d414c7e
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File

30
timm/models/metaformers.py
Unescape View File

@ -772,7 +772,7 @@ default_cfgs = generate_default_cfgs({
classifier='head.fc.fc2'),
'convformer_s18.sail_in22k_ft_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s18_384_in21ft1k.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'convformer_s18.sail_in22k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s18_in21k.pth',
classifier='head.fc.fc2', num_classes=21841),
@ -782,13 +782,13 @@ default_cfgs = generate_default_cfgs({
classifier='head.fc.fc2'),
'convformer_s36.sail_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s36_384.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'convformer_s36.sail_in22k_ft_in1k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s36_in21ft1k.pth',
classifier='head.fc.fc2'),
'convformer_s36.sail_in22k_ft_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s36_384_in21ft1k.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'convformer_s36.sail_in22k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s36_in21k.pth',
classifier='head.fc.fc2', num_classes=21841),
@ -798,13 +798,13 @@ default_cfgs = generate_default_cfgs({
classifier='head.fc.fc2'),
'convformer_m36.sail_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_m36_384.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'convformer_m36.sail_in22k_ft_in1k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_m36_in21ft1k.pth',
classifier='head.fc.fc2'),
'convformer_m36.sail_in22k_ft_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_m36_384_in21ft1k.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'convformer_m36.sail_in22k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_m36_in21k.pth',
classifier='head.fc.fc2', num_classes=21841),
@ -814,13 +814,13 @@ default_cfgs = generate_default_cfgs({
classifier='head.fc.fc2'),
'convformer_b36.sail_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_b36_384.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'convformer_b36.sail_in22k_ft_in1k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_b36_in21ft1k.pth',
classifier='head.fc.fc2'),
'convformer_b36.sail_in22k_ft_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_b36_384_in21ft1k.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'convformer_b36.sail_in22k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_b36_in21k.pth',
classifier='head.fc.fc2', num_classes=21841),
@ -831,13 +831,13 @@ default_cfgs = generate_default_cfgs({
classifier='head.fc.fc2'),
'caformer_s18.sail_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_s18_384.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'caformer_s18.sail_in22k_ft_in1k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_s18_in21ft1k.pth',
classifier='head.fc.fc2'),
'caformer_s18.sail_in22k_ft_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_s18_384_in21ft1k.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'caformer_s18.sail_in22k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_s18_in21k.pth',
classifier='head.fc.fc2', num_classes=21841),
@ -847,13 +847,13 @@ default_cfgs = generate_default_cfgs({
classifier='head.fc.fc2'),
'caformer_s36.sail_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_s36_384.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'caformer_s36.sail_in22k_ft_in1k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_s36_in21ft1k.pth',
classifier='head.fc.fc2'),
'caformer_s36.sail_in22k_ft_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_s36_384_in21ft1k.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'caformer_s36.sail_in22k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_s36_in21k.pth',
classifier='head.fc.fc2', num_classes=21841),
@ -863,13 +863,13 @@ default_cfgs = generate_default_cfgs({
classifier='head.fc.fc2'),
'caformer_m36.sail_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_m36_384.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'caformer_m36.sail_in22k_ft_in1k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_m36_in21ft1k.pth',
classifier='head.fc.fc2'),
'caformer_m36.sail_in22k_ft_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_m36_384_in21ft1k.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'caformer_m36.sail_in22k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_m36_in21k.pth',
classifier='head.fc.fc2', num_classes=21841),
@ -879,13 +879,13 @@ default_cfgs = generate_default_cfgs({
classifier='head.fc.fc2'),
'caformer_b36.sail_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_b36_384.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'caformer_b36.sail_in22k_ft_in1k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_b36_in21ft1k.pth',
classifier='head.fc.fc2'),
'caformer_b36.sail_in22k_ft_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_b36_384_in21ft1k.pth',
classifier='head.fc.fc2', input_size=(3, 384, 384)),
classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)),
'caformer_b36.sail_in22k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/caformer/caformer_b36_in21k.pth',
classifier='head.fc.fc2', num_classes=21841),

Write Preview
Loading…
Cancel
Save