Add TorchCodec video backend

2026-08-23 16:41:05 +08:00
parent b875a7512e
commit c844673984
5 changed files with 112 additions and 17 deletions
@@ -95,7 +95,11 @@ def parse_args(argv: Sequence[str] | None = None) -> argparse.Namespace:
            "original maximum-length padding"
        ),
    )
-    parser.add_argument("--decoder", choices=("decord", "pynvcodec"), default="decord")
+    parser.add_argument(
+        "--decoder",
+        choices=("decord", "torchcodec", "pynvcodec"),
+        default="decord",
+    )
    parser.add_argument("--decoder-gpu-id", type=int, default=0)
    parser.add_argument("--decoder-threads", type=int, default=1)
    parser.add_argument("--max-open-readers", type=int, default=1)