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():