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

finalizing

Browse Source
pull/122/head
talrid 5 years ago
parent 6209146738
commit 8a63c1add8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
timm/models/tresnet.py
Unescape 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:

Write Preview
Loading…
Cancel
Save