From a7ecb62833221bf82f701020fe55a4b221e0386e Mon Sep 17 00:00:00 2001 From: Dmitriy Genzel Date: Thu, 17 Jun 2021 10:02:29 -0700 Subject: [PATCH] Enable analytics logging for MNIST and CIFAR Summary: Enable analytics logging for MNIST and CIFAR Reviewed By: colin2328, fmassa Differential Revision: D29183132 fbshipit-source-id: 67d9f6881354f87659fdec4183b22a1be3ec0eb6 --- torchvision/datasets/cifar.py | 2 ++ torchvision/datasets/mnist.py | 1 + 2 files changed, 3 insertions(+) diff --git a/torchvision/datasets/cifar.py b/torchvision/datasets/cifar.py index 9d939326c76..47b2bd41fb0 100644 --- a/torchvision/datasets/cifar.py +++ b/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 diff --git a/torchvision/datasets/mnist.py b/torchvision/datasets/mnist.py index e356f17dd1b..edd2185c984 100644 --- a/torchvision/datasets/mnist.py +++ b/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():