Remove some outdated comments, botnet networks working great now.

pull/913/head
Ross Wightman 3 years ago
parent 6ed4cdccca
commit cd34913278

@ -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)

Loading…
Cancel
Save