[FBcode->GH] Enable analytics logging for MNIST and CIFAR (#4090)

Summary: Enable analytics logging for MNIST and CIFAR

Reviewed By: colin2328, fmassa

Differential Revision: D29183132

fbshipit-source-id: 67d9f6881354f87659fdec4183b22a1be3ec0eb6

Co-authored-by: Dmitriy Genzel <[email protected]>

Author: NicolasHug

torchvision/datasets/cifar.py

@@ -3,6 +3,7 @@
 import os.path
 import numpy as np
 import pickle
+import torch
 from typing import Any, Callable, Optional, Tuple
 
 from .vision import VisionDataset
@@ -58,6 +59,7 @@ def __init__(
 
         super(CIFAR10, self).__init__(root, transform=transform,
                                       target_transform=target_transform)
+        torch._C._log_api_usage_once(f"torchvision.datasets.{self.__class__.__name__}")
 
         self.train = train  # training set or test set
 

torchvision/datasets/mnist.py

@@ -77,6 +77,7 @@ def __init__(
     ) -> None:
         super(MNIST, self).__init__(root, transform=transform,
                                     target_transform=target_transform)
+        torch._C._log_api_usage_once(f"torchvision.datasets.{self.__class__.__name__}")
         self.train = train  # training set or test set
 
         if self._check_legacy_exist():