From 64495505b7bbf5438672d53804efaaa634bad710 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 3 Oct 2021 17:31:39 -0700 Subject: [PATCH] Add updated lambda resnet26 and botnet26 checkpoints with fixes applied --- timm/models/byoanet.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/timm/models/byoanet.py b/timm/models/byoanet.py index f58b724c..3c43378a 100644 --- a/timm/models/byoanet.py +++ b/timm/models/byoanet.py @@ -34,8 +34,8 @@ def _cfg(url='', **kwargs): default_cfgs = { # GPU-Efficient (ResNet) weights 'botnet26t_256': _cfg( - url='', - fixed_input_size=True, input_size=(3, 256, 256), pool_size=(8, 8)), + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-attn-weights/botnet26t_c1_256-167a0e9f.pth', + fixed_input_size=True, input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=0.95), 'botnet50ts_256': _cfg( url='', fixed_input_size=True, input_size=(3, 256, 256), pool_size=(8, 8)), @@ -58,13 +58,13 @@ default_cfgs = { input_size=(3, 256, 256), pool_size=(8, 8), min_input_size=(3, 256, 256), crop_pct=0.94), 'lambda_resnet26t': _cfg( - url='', - min_input_size=(3, 128, 128), input_size=(3, 256, 256), pool_size=(8, 8)), + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-attn-weights/lambda_resnet26t_a2h_256-25ded63d.pth', + min_input_size=(3, 128, 128), input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=0.95), 'lambda_resnet50ts': _cfg( url='', min_input_size=(3, 128, 128), input_size=(3, 256, 256), pool_size=(8, 8)), 'lambda_resnet26rpt_256': _cfg( - url='', + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-attn-weights/lambda_resnet26rpt_a2h_256-482adad8.pth', fixed_input_size=True, input_size=(3, 256, 256), pool_size=(8, 8)), }