typo

pytorch · Dec 9, 2021 · 2c14700 · 2c14700
1 parent f2d9fbc
commit 2c14700
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchvision/prototype/models/optical_flow/raft.py b/torchvision/prototype/models/optical_flow/raft.py
@@ -86,7 +86,7 @@ class Raft_Large_Weights(WeightsEnum):
         meta={
             **_COMMON_META,
             "recipe": "https://github.com/princeton-vl/RAFT",
-            "kitti_train_f1-all": 5.10,
+            "kitti_test_f1-all": 5.10,
         },
     )
 
@@ -100,7 +100,7 @@ class Raft_Large_Weights(WeightsEnum):
         meta={
             **_COMMON_META,
             "recipe": "https://github.com/pytorch/vision/tree/main/references/optical_flow",
-            "kitti_train_f1-all": 5.19,
+            "kitti_test_f1-all": 5.19,
         },
     )