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

Fix typos RepGhost models

Browse Source
This commit is contained in:
Chengpeng Chen 2023-08-16 14:27:45 +08:00
parent d1d0193615
commit e7f97cb5ce
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
timm/models/repghost.py
View File

@ -441,19 +441,19 @@ def repghostnet_111(pretrained=False, **kwargs) -> RepGhostNet:
@register_model @register_model
def repghostnet_130(pretrained=False, **kwargs) -> RepGhostNet: def repghostnet_130(pretrained=False, **kwargs) -> RepGhostNet:
""" RepGhostNet-1.3x """ """ RepGhostNet-1.3x """
model = _create_repghostnet('ghostnet_130', width=1.3, pretrained=pretrained, **kwargs) model = _create_repghostnet('repghostnet_130', width=1.3, pretrained=pretrained, **kwargs)
return model return model
@register_model @register_model
def repghostnet_150(pretrained=False, **kwargs) -> RepGhostNet: def repghostnet_150(pretrained=False, **kwargs) -> RepGhostNet:
""" RepGhostNet-1.5x """ """ RepGhostNet-1.5x """
model = _create_repghostnet('ghostnet_150', width=1.5, pretrained=pretrained, **kwargs) model = _create_repghostnet('repghostnet_150', width=1.5, pretrained=pretrained, **kwargs)
return model return model
@register_model @register_model
def repghostnet_200(pretrained=False, **kwargs) -> RepGhostNet: def repghostnet_200(pretrained=False, **kwargs) -> RepGhostNet:
""" RepGhostNet-2.0x """ """ RepGhostNet-2.0x """
model = _create_repghostnet('ghostnet_200', width=2.0, pretrained=pretrained, **kwargs) model = _create_repghostnet('repghostnet_200', width=2.0, pretrained=pretrained, **kwargs)
return model return model