Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "3ebbaf7c96801271f9e6c21400033b6aa5ffcf29"
Unverified Commit 22a23e8d authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix download link for raft_small (#5078)

parent 2508a020
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
references/optical_flow/README.md
View file @ 22a23e8d
...@@ -62,4 +62,4 @@ You can also evaluate on Kitti train: ...@@ -62,4 +62,4 @@ You can also evaluate on Kitti train:
``` ```
torchrun --nproc_per_node 1 --nnodes 1 train.py --val-dataset kitti --batch-size 1 --dataset-root $dataset_root --model raft_large --pretrained torchrun --nproc_per_node 1 --nnodes 1 train.py --val-dataset kitti --batch-size 1 --dataset-root $dataset_root --model raft_large --pretrained
Kitti val epe: 4.7968 1px: 0.6388 3px: 0.8197 5px: 0.8661 per_image_epe: 4.5118 f1: 16.0679 Kitti val epe: 4.7968 1px: 0.6388 3px: 0.8197 5px: 0.8661 per_image_epe: 4.5118 f1: 16.0679
``` ```
\ No newline at end of file
torchvision/prototype/models/optical_flow/raft.py
View file @ 22a23e8d
...@@ -97,11 +97,11 @@ class Raft_Small_Weights(WeightsEnum): ...@@ -97,11 +97,11 @@ class Raft_Small_Weights(WeightsEnum):
) )
C_T_V2 = Weights( C_T_V2 = Weights(
# Chairs + Things # Chairs + Things
url="https://github.com/pytorch/vision/tree/main/references/optical_flow", url="https://download.pytorch.org/models/raft_small_C_T_V2-01064c6d.pth",
transforms=RaftEval, transforms=RaftEval,
meta={ meta={
**_COMMON_META, **_COMMON_META,
"recipe": "https://github.com/princeton-vl/RAFT", "recipe": "https://github.com/pytorch/vision/tree/main/references/optical_flow",
"sintel_train_cleanpass_epe": 1.9901, "sintel_train_cleanpass_epe": 1.9901,
"sintel_train_finalpass_epe": 3.2831, "sintel_train_finalpass_epe": 3.2831,
"kitti_train_per_image_epe": 7.5978, "kitti_train_per_image_epe": 7.5978,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment