From 4015272a4e273e2b570d2621b8eaf8b286ef14e4 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Tue, 5 Jul 2022 09:11:47 +0100 Subject: [PATCH 1/2] Make `_log_api_usage_once` work for overwritten classes --- torchvision/utils.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/torchvision/utils.py b/torchvision/utils.py index 7eb4f4a3d7a..f5499aaa389 100644 --- a/torchvision/utils.py +++ b/torchvision/utils.py @@ -562,9 +562,10 @@ def _log_api_usage_once(obj: Any) -> None: Args: obj (class instance or method): an object to extract info from. """ - if not obj.__module__.startswith("torchvision"): - return + module = obj.__module__ + if not module.startswith("torchvision"): + module = f"torchvision.fb.{module}" name = obj.__class__.__name__ if isinstance(obj, FunctionType): name = obj.__name__ - torch._C._log_api_usage_once(f"{obj.__module__}.{name}") + torch._C._log_api_usage_once(f"{module}.{name}") From 0edaaa867cbfc6e1c83105d21853ee174ec7ea18 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Tue, 5 Jul 2022 09:59:50 +0100 Subject: [PATCH 2/2] Change module prefix to `torchvision.internal` --- torchvision/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/utils.py b/torchvision/utils.py index f5499aaa389..4c8e897f40a 100644 --- a/torchvision/utils.py +++ b/torchvision/utils.py @@ -564,7 +564,7 @@ def _log_api_usage_once(obj: Any) -> None: """ module = obj.__module__ if not module.startswith("torchvision"): - module = f"torchvision.fb.{module}" + module = f"torchvision.internal.{module}" name = obj.__class__.__name__ if isinstance(obj, FunctionType): name = obj.__name__