Unverified Commit 7a0184aa authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

[FBcode->GH] Moving logging on base class. (#4319)

parent 32f95c77
......@@ -59,7 +59,6 @@ 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
......
......@@ -77,7 +77,6 @@ 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():
......
......@@ -31,6 +31,7 @@ class VisionDataset(data.Dataset):
transform: Optional[Callable] = None,
target_transform: Optional[Callable] = None,
) -> None:
torch._C._log_api_usage_once(f"torchvision.datasets.{self.__class__.__name__}")
if isinstance(root, torch._six.string_classes):
root = os.path.expanduser(root)
self.root = root
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment