Fixing typing.

Author: datumbox

torchvision/prototype/models/_api.py

@@ -30,7 +30,7 @@ class WeightEntry:
     transforms: Callable
     meta: Dict[str, Any]
 
-    def state_dict(self, progress: bool) -> Dict[str, Any]:
+    def state_dict(self, progress: bool) -> OrderedDict:
         return load_state_dict_from_url(self.url, progress=progress)