File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
torchvision/prototype/models/optical_flow Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -62,4 +62,4 @@ You can also evaluate on Kitti train:
6262```
6363torchrun --nproc_per_node 1 --nnodes 1 train.py --val-dataset kitti --batch-size 1 --dataset-root $dataset_root --model raft_large --pretrained
6464Kitti val epe: 4.7968 1px: 0.6388 3px: 0.8197 5px: 0.8661 per_image_epe: 4.5118 f1: 16.0679
65- ```
65+ ```
Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments