From cd34913278f8511ba53492ed186b4e08f890add6 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 11 Oct 2021 22:43:41 -0700 Subject: [PATCH] Remove some outdated comments, botnet networks working great now. --- timm/models/byoanet.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/timm/models/byoanet.py b/timm/models/byoanet.py index 313af3e2..d296d4ba 100644 --- a/timm/models/byoanet.py +++ b/timm/models/byoanet.py @@ -294,7 +294,6 @@ def _create_byoanet(variant, cfg_variant=None, pretrained=False, **kwargs): @register_model def botnet26t_256(pretrained=False, **kwargs): """ Bottleneck Transformer w/ ResNet26-T backbone. - NOTE: this isn't performing well, may remove """ kwargs.setdefault('img_size', 256) return _create_byoanet('botnet26t_256', 'botnet26t', pretrained=pretrained, **kwargs) @@ -303,7 +302,6 @@ def botnet26t_256(pretrained=False, **kwargs): @register_model def botnet50ts_256(pretrained=False, **kwargs): """ Bottleneck Transformer w/ ResNet50-T backbone, silu act. - NOTE: this isn't performing well, may remove """ kwargs.setdefault('img_size', 256) return _create_byoanet('botnet50ts_256', 'botnet50ts', pretrained=pretrained, **kwargs) @@ -312,7 +310,6 @@ def botnet50ts_256(pretrained=False, **kwargs): @register_model def eca_botnext26ts_256(pretrained=False, **kwargs): """ Bottleneck Transformer w/ ResNet26-T backbone, silu act. - NOTE: this isn't performing well, may remove """ kwargs.setdefault('img_size', 256) return _create_byoanet('eca_botnext26ts_256', 'eca_botnext26ts', pretrained=pretrained, **kwargs) @@ -385,6 +382,6 @@ def haloregnetz_b(pretrained=False, **kwargs): @register_model def trionet50ts_256(pretrained=False, **kwargs): - """ HaloNet w/ a ResNet50-t backbone, silu act. Halo attention in final two stages + """ TrioNet """ return _create_byoanet('trionet50ts_256', 'trionet50ts', pretrained=pretrained, **kwargs)