Extend _log_api_usage_once to work for overwritten classes (#6237)

* Make `_log_api_usage_once` work for overwritten classes

* Change module prefix to `torchvision.internal`

Author: datumbox

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}")