We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
handle_legacy_interface
1 parent c780be5 commit d16d5b0Copy full SHA for d16d5b0
torchvision/prototype/models/_utils.py
@@ -43,6 +43,7 @@ def inner_wrapper(*args: Any, **kwargs: Any) -> M:
43
if (
44
(weights_param not in kwargs and pretrained_param not in kwargs)
45
or isinstance(weights_arg, WeightsEnum)
46
+ or (isinstance(weights_arg, str) and weights_arg != "legacy")
47
or weights_arg is None
48
):
49
continue
0 commit comments