diff --git a/torchvision/utils.py b/torchvision/utils.py index 7eb4f4a3d7a..4c8e897f40a 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.internal.{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}")