Skip to content

Commit dffcc9e

Browse files
NicolasHugfacebook-github-bot
authored andcommitted
[fbsync] Fix download link for raft_small (#5078)
Reviewed By: fmassa Differential Revision: D33185000 fbshipit-source-id: be7cfb0195df5f5c02ed35018b60086b4d077cc1
1 parent fb58270 commit dffcc9e

File tree

2 files changed

+3
-3
lines changed
  • references/optical_flow
  • torchvision/prototype/models/optical_flow

2 files changed

+3
-3
lines changed

references/optical_flow/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,4 +62,4 @@ You can also evaluate on Kitti train:
6262
```
6363
torchrun --nproc_per_node 1 --nnodes 1 train.py --val-dataset kitti --batch-size 1 --dataset-root $dataset_root --model raft_large --pretrained
6464
Kitti val epe: 4.7968 1px: 0.6388 3px: 0.8197 5px: 0.8661 per_image_epe: 4.5118 f1: 16.0679
65-
```
65+
```

torchvision/prototype/models/optical_flow/raft.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,11 +97,11 @@ class Raft_Small_Weights(WeightsEnum):
9797
)
9898
C_T_V2 = Weights(
9999
# Chairs + Things
100-
url="https://github.com/pytorch/vision/tree/main/references/optical_flow",
100+
url="https://download.pytorch.org/models/raft_small_C_T_V2-01064c6d.pth",
101101
transforms=RaftEval,
102102
meta={
103103
**_COMMON_META,
104-
"recipe": "https://github.com/princeton-vl/RAFT",
104+
"recipe": "https://github.com/pytorch/vision/tree/main/references/optical_flow",
105105
"sintel_train_cleanpass_epe": 1.9901,
106106
"sintel_train_finalpass_epe": 3.2831,
107107
"kitti_train_per_image_epe": 7.5978,

0 commit comments

Comments
 (0)