Update torchvision/prototype/models/optical_flow/raft.py

Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
pytorch · Dec 9, 2021 · d8431fc · d8431fc
1 parent 3c07750
commit d8431fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/prototype/models/optical_flow/raft.py b/torchvision/prototype/models/optical_flow/raft.py
@@ -97,7 +97,7 @@ class Raft_Small_Weights(WeightsEnum):
     )
     C_T_V2 = Weights(
         # Chairs + Things
-        url="https://download.pytorch.org/models/raft_small_C_T_V2-01064c6d.pth",
+        url="https://github.com/pytorch/vision/tree/main/references/optical_flow",
         transforms=RaftEval,
         meta={
             **_COMMON_META,