[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 <dgenzel@fb.com>

torchvision/datasets/cifar.py

@@ -3,6 +3,7 @@ import os
 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 @@ class CIFAR10(VisionDataset):
 
         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 @@ class MNIST(VisionDataset):
     ) -> 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():