Expose all options

adamjstewart · adamjstewart · commit 9acc3ee97851 · 2023-03-20T09:29:03.000-06:00
diff --git a/torchvision/models/_api.py b/torchvision/models/_api.py
@@ -85,8 +85,8 @@ def verify(cls, obj: Any) -> Any:
                 )
         return obj
 
-    def get_state_dict(self, progress: bool) -> Mapping[str, Any]:
-        return load_state_dict_from_url(self.url, progress=progress, check_hash=True)
+    def get_state_dict(self, *args: Any, **kwargs: Any) -> Mapping[str, Any]:
+        return load_state_dict_from_url(self.url, *args, **kwargs)
 
     def __repr__(self) -> str:
         return f"{self.__class__.__name__}.{self._name_}"

Original file line number	Diff line number	Diff line change
`@@ -85,8 +85,8 @@ def verify(cls, obj: Any) -> Any:`
`85`	`85`	`)`
`86`	`86`	`return obj`
`87`	`87`
`88`		`- def get_state_dict(self, progress: bool) -> Mapping[str, Any]:`
`89`		`- return load_state_dict_from_url(self.url, progress=progress, check_hash=True)`
	`88`	`+ def get_state_dict(self, args: Any, *kwargs: Any) -> Mapping[str, Any]:`
	`89`	`+ return load_state_dict_from_url(self.url, args, *kwargs)`
`90`	`90`
`91`	`91`	`def __repr__(self) -> str:`
`92`	`92`	`return f"{self.__class__.__name__}.{self._name_}"`