Temporary fix for PyTorch 403 error.

n2cholas · Aug 11, 2021 · 43f4119 · 43f4119
1 parent 2eb47a4
commit 43f4119
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/jax_resnet/pretrained.py b/jax_resnet/pretrained.py
@@ -10,6 +10,9 @@
 
 try:
     import torch
+
+    # https://github.com/pytorch/vision/issues/4156#issuecomment-894768539
+    torch.hub._validate_not_a_forked_repo = lambda *_: True  # type: ignore
     torch_exists = True
 except ImportError:
     torch_exists = False