diff --git a/torchvision/prototype/models/optical_flow/raft.py b/torchvision/prototype/models/optical_flow/raft.py index f0079ab344b..ca4ae90927e 100644 --- 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,