Unverified Commit 11480973 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

[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: default avatarDmitriy Genzel <dgenzel@fb.com>
parent 655a2ede
...@@ -3,6 +3,7 @@ import os ...@@ -3,6 +3,7 @@ import os
import os.path import os.path
import numpy as np import numpy as np
import pickle import pickle
import torch
from typing import Any, Callable, Optional, Tuple from typing import Any, Callable, Optional, Tuple
from .vision import VisionDataset from .vision import VisionDataset
...@@ -58,6 +59,7 @@ class CIFAR10(VisionDataset): ...@@ -58,6 +59,7 @@ class CIFAR10(VisionDataset):
super(CIFAR10, self).__init__(root, transform=transform, super(CIFAR10, self).__init__(root, transform=transform,
target_transform=target_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 self.train = train # training set or test set
......
...@@ -77,6 +77,7 @@ class MNIST(VisionDataset): ...@@ -77,6 +77,7 @@ class MNIST(VisionDataset):
) -> None: ) -> None:
super(MNIST, self).__init__(root, transform=transform, super(MNIST, self).__init__(root, transform=transform,
target_transform=target_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 self.train = train # training set or test set
if self._check_legacy_exist(): if self._check_legacy_exist():
......
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