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

finalizing

Browse Source
This commit is contained in:
talrid 2020-04-12 19:08:35 +03:00
parent 6209146738
commit 8a63c1add8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/tresnet.py
View File

@ -38,7 +38,7 @@ default_cfgs = {
'tresnet_l': 'tresnet_l':
_cfg(url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/tresnet/tresnet_l_81_5.pth'), _cfg(url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/tresnet/tresnet_l_81_5.pth'),
'tresnet_xl': 'tresnet_xl':
_cfg(url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/tresnet/tresnet_xl_82_1.pth') _cfg(url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/tresnet/tresnet_xl_82_0.pth')
} }
@ -284,7 +284,7 @@ def tresnet_l(pretrained=False, num_classes=1000, in_chans=3, **kwargs):
@register_model @register_model
def tresnet_xl(pretrained=False, num_classes=1000, in_chans=3, **kwargs): def tresnet_xl(pretrained=False, num_classes=1000, in_chans=3, **kwargs):
default_cfg = default_cfgs['tresnet_l'] default_cfg = default_cfgs['tresnet_xl']
model = TResNet(layers=[4, 5, 24, 3], num_classes=num_classes, in_chans=in_chans, width_factor=1.3) model = TResNet(layers=[4, 5, 24, 3], num_classes=num_classes, in_chans=in_chans, width_factor=1.3)
model.default_cfg = default_cfg model.default_cfg = default_cfg
if pretrained: if pretrained: