use decorator to log API usage

Author: kazhang

torchvision/datasets/vision.py

@@ -28,14 +28,14 @@ class VisionDataset(data.Dataset):
 
     _repr_indent = 4
 
+    @_log_api_usage_once
     def __init__(
         self,
         root: str,
         transforms: Optional[Callable] = None,
         transform: Optional[Callable] = None,
         target_transform: Optional[Callable] = None,
     ) -> None:
-        _log_api_usage_once(self)
         if isinstance(root, torch._six.string_classes):
             root = os.path.expanduser(root)
         self.root = root

torchvision/models/alexnet.py

@@ -16,9 +16,9 @@
 
 
 class AlexNet(nn.Module):
+    @_log_api_usage_once
     def __init__(self, num_classes: int = 1000, dropout: float = 0.5) -> None:
         super().__init__()
-        _log_api_usage_once(self)
         self.features = nn.Sequential(
             nn.Conv2d(3, 64, kernel_size=11, stride=4, padding=2),
             nn.ReLU(inplace=True),

torchvision/models/densenet.py

@@ -151,6 +151,7 @@ class DenseNet(nn.Module):
           but slower. Default: *False*. See `"paper" <https://arxiv.org/pdf/1707.06990.pdf>`_.
     """
 
+    @_log_api_usage_once
     def __init__(
         self,
         growth_rate: int = 32,
@@ -161,9 +162,7 @@ def __init__(
         num_classes: int = 1000,
         memory_efficient: bool = False,
     ) -> None:
-
         super().__init__()
-        _log_api_usage_once(self)
 
         # First convolution
         self.features = nn.Sequential(

torchvision/models/detection/generalized_rcnn.py

@@ -25,9 +25,9 @@ class GeneralizedRCNN(nn.Module):
             the model
     """
 
+    @_log_api_usage_once
     def __init__(self, backbone: nn.Module, rpn: nn.Module, roi_heads: nn.Module, transform: nn.Module) -> None:
         super().__init__()
-        _log_api_usage_once(self)
         self.transform = transform
         self.backbone = backbone
         self.rpn = rpn

torchvision/models/detection/retinanet.py

@@ -316,6 +316,7 @@ class RetinaNet(nn.Module):
         "proposal_matcher": det_utils.Matcher,
     }
 
+    @_log_api_usage_once
     def __init__(
         self,
         backbone,
@@ -337,7 +338,6 @@ def __init__(
         topk_candidates=1000,
     ):
         super().__init__()
-        _log_api_usage_once(self)
 
         if not hasattr(backbone, "out_channels"):
             raise ValueError(

torchvision/models/detection/ssd.py

@@ -165,6 +165,7 @@ class SSD(nn.Module):
         "proposal_matcher": det_utils.Matcher,
     }
 
+    @_log_api_usage_once
     def __init__(
         self,
         backbone: nn.Module,
@@ -182,7 +183,6 @@ def __init__(
         positive_fraction: float = 0.25,
     ):
         super().__init__()
-        _log_api_usage_once(self)
 
         self.backbone = backbone
 

torchvision/models/detection/ssdlite.py

@@ -111,6 +111,7 @@ def __init__(self, in_channels: List[int], num_anchors: List[int], norm_layer: C
 
 
 class SSDLiteFeatureExtractorMobileNet(nn.Module):
+    @_log_api_usage_once
     def __init__(
         self,
         backbone: nn.Module,
@@ -120,7 +121,6 @@ def __init__(
         min_depth: int = 16,
     ):
         super().__init__()
-        _log_api_usage_once(self)
 
         assert not backbone[c4_pos].use_res_connect
         self.features = nn.Sequential(

torchvision/models/efficientnet.py

@@ -148,6 +148,7 @@ def forward(self, input: Tensor) -> Tensor:
 
 
 class EfficientNet(nn.Module):
+    @_log_api_usage_once
     def __init__(
         self,
         inverted_residual_setting: List[MBConvConfig],
@@ -170,7 +171,6 @@ def __init__(
             norm_layer (Optional[Callable[..., nn.Module]]): Module specifying the normalization layer to use
         """
         super().__init__()
-        _log_api_usage_once(self)
 
         if not inverted_residual_setting:
             raise ValueError("The inverted_residual_setting should not be empty")

torchvision/models/googlenet.py

@@ -28,6 +28,7 @@
 class GoogLeNet(nn.Module):
     __constants__ = ["aux_logits", "transform_input"]
 
+    @_log_api_usage_once
     def __init__(
         self,
         num_classes: int = 1000,
@@ -39,7 +40,6 @@ def __init__(
         dropout_aux: float = 0.7,
     ) -> None:
         super().__init__()
-        _log_api_usage_once(self)
         if blocks is None:
             blocks = [BasicConv2d, Inception, InceptionAux]
         if init_weights is None:

torchvision/models/inception.py

@@ -27,6 +27,7 @@
 
 
 class Inception3(nn.Module):
+    @_log_api_usage_once
     def __init__(
         self,
         num_classes: int = 1000,
@@ -37,7 +38,6 @@ def __init__(
         dropout: float = 0.5,
     ) -> None:
         super().__init__()
-        _log_api_usage_once(self)
         if inception_blocks is None:
             inception_blocks = [BasicConv2d, InceptionA, InceptionB, InceptionC, InceptionD, InceptionE, InceptionAux]
         if init_weights is None: