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-18 10:41:22 -08:00
parent 924a64f051
commit 93ed33bc43
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

@ -751,7 +751,8 @@ default_cfgs = generate_default_cfgs({
'convformer_s18.sail_in1k': _cfg( 'convformer_s18.sail_in1k': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s18.pth'), url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s18.pth',
classifier='head.fc.fc2'),
'convformer_s18.sail_in1k_384': _cfg( 'convformer_s18.sail_in1k_384': _cfg(
url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s18_384.pth', url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s18_384.pth',
input_size=(3, 384, 384)), input_size=(3, 384, 384)),