Fix broken links for NTS-Net

Swapnil2095 · May 28, 2019 · 612da51 · 612da51
1 parent 0d21eeb
commit 612da51
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/chainer_/chainercv2/models/ntsnet_cub.py b/chainer_/chainercv2/models/ntsnet_cub.py
@@ -368,7 +368,7 @@ def ntsnet_cub(pretrained_backbone=False, aux=True, **kwargs):
     """
     backbone = resnet50b(pretrained=pretrained_backbone).features
     del backbone.final_pool
-    return get_ntsnet(backbone=backbone, aux=aux, model_name="ntsnet", **kwargs)
+    return get_ntsnet(backbone=backbone, aux=aux, model_name="ntsnet_cub", **kwargs)
 
 
 def _test():

diff --git a/gluon/gluoncv2/models/ntsnet_cub.py b/gluon/gluoncv2/models/ntsnet_cub.py
@@ -371,7 +371,7 @@ def ntsnet_cub(pretrained_backbone=False, aux=True, **kwargs):
         Location for keeping the model parameters.
     """
     backbone = resnet50b(pretrained=pretrained_backbone).features[:-1]
-    return get_ntsnet(backbone=backbone, aux=aux, model_name="ntsnet", **kwargs)
+    return get_ntsnet(backbone=backbone, aux=aux, model_name="ntsnet_cub", **kwargs)
 
 
 def _test():

diff --git a/pytorch/pytorchcv/models/ntsnet_cub.py b/pytorch/pytorchcv/models/ntsnet_cub.py
@@ -371,7 +371,7 @@ def ntsnet_cub(pretrained_backbone=False, aux=True, **kwargs):
     """
     backbone = resnet50b(pretrained=pretrained_backbone).features
     del backbone[-1]
-    return get_ntsnet(backbone=backbone, aux=aux, model_name="ntsnet", **kwargs)
+    return get_ntsnet(backbone=backbone, aux=aux, model_name="ntsnet_cub", **kwargs)
 
 
 def _calc_width(net):