[fbsync] Fix mypy job failing (#5970)

Reviewed By: YosuaMichael

Differential Revision: D36281601

fbshipit-source-id: 33d2ba14754d6787b9a7b1812e17b081917eb4a1

Author: datumbox

torchvision/models/_api.py

@@ -1,7 +1,6 @@
 import importlib
 import inspect
 import sys
-from collections import OrderedDict
 from dataclasses import dataclass, fields
 from inspect import signature
 from typing import Any, Callable, Dict, cast
@@ -60,7 +59,7 @@ def verify(cls, obj: Any) -> Any:
                 )
         return obj
 
-    def get_state_dict(self, progress: bool) -> OrderedDict:
+    def get_state_dict(self, progress: bool) -> Dict[str, Any]:
         return load_state_dict_from_url(self.url, progress=progress)
 
     def __repr__(self) -> str: