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

Remove some outdated comments, botnet networks working great now.

Browse Source
This commit is contained in:
Ross Wightman 2021-10-11 22:43:41 -07:00
parent 6ed4cdccca
commit cd34913278
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
timm/models/byoanet.py
View File

@ -294,7 +294,6 @@ def _create_byoanet(variant, cfg_variant=None, pretrained=False, **kwargs):
@register_model @register_model
def botnet26t_256(pretrained=False, **kwargs): def botnet26t_256(pretrained=False, **kwargs):
""" Bottleneck Transformer w/ ResNet26-T backbone. """ Bottleneck Transformer w/ ResNet26-T backbone.
NOTE: this isn't performing well, may remove
""" """
kwargs.setdefault('img_size', 256) kwargs.setdefault('img_size', 256)
return _create_byoanet('botnet26t_256', 'botnet26t', pretrained=pretrained, **kwargs) return _create_byoanet('botnet26t_256', 'botnet26t', pretrained=pretrained, **kwargs)
@ -303,7 +302,6 @@ def botnet26t_256(pretrained=False, **kwargs):
@register_model @register_model
def botnet50ts_256(pretrained=False, **kwargs): def botnet50ts_256(pretrained=False, **kwargs):
""" Bottleneck Transformer w/ ResNet50-T backbone, silu act. """ Bottleneck Transformer w/ ResNet50-T backbone, silu act.
NOTE: this isn't performing well, may remove
""" """
kwargs.setdefault('img_size', 256) kwargs.setdefault('img_size', 256)
return _create_byoanet('botnet50ts_256', 'botnet50ts', pretrained=pretrained, **kwargs) return _create_byoanet('botnet50ts_256', 'botnet50ts', pretrained=pretrained, **kwargs)
@ -312,7 +310,6 @@ def botnet50ts_256(pretrained=False, **kwargs):
@register_model @register_model
def eca_botnext26ts_256(pretrained=False, **kwargs): def eca_botnext26ts_256(pretrained=False, **kwargs):
""" Bottleneck Transformer w/ ResNet26-T backbone, silu act. """ Bottleneck Transformer w/ ResNet26-T backbone, silu act.
NOTE: this isn't performing well, may remove
""" """
kwargs.setdefault('img_size', 256) kwargs.setdefault('img_size', 256)
return _create_byoanet('eca_botnext26ts_256', 'eca_botnext26ts', pretrained=pretrained, **kwargs) return _create_byoanet('eca_botnext26ts_256', 'eca_botnext26ts', pretrained=pretrained, **kwargs)
@ -385,6 +382,6 @@ def haloregnetz_b(pretrained=False, **kwargs):
@register_model @register_model
def trionet50ts_256(pretrained=False, **kwargs): 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) return _create_byoanet('trionet50ts_256', 'trionet50ts', pretrained=pretrained, **kwargs)