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

NicolasHug · datumbox · web-flow · commit d8431fc9c5bf · 2021-12-09T11:34:09.000Z
Co-authored-by: Vasilis Vryniotis &lt;datumbox@users.noreply.github.com&gt;
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,

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ class Raft_Small_Weights(WeightsEnum):`
`97`	`97`	`)`
`98`	`98`	`C_T_V2 = Weights(`
`99`	`99`	`# Chairs + Things`
`100`		`- url="https://download.pytorch.org/models/raft_small_C_T_V2-01064c6d.pth",`
	`100`	`+ url="https://github.com/pytorch/vision/tree/main/references/optical_flow",`
`101`	`101`	`transforms=RaftEval,`
`102`	`102`	`meta={`
`103`	`103`	`**_COMMON_META,`