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
@@ -113,7 +113,11 @@ def parse_args(argv: Sequence[str] | None = None) -> argparse.Namespace:
        help="Maximum producer-ready CPU batches queued ahead of training",
    )
    parser.add_argument("--frame-cache-size", type=int, default=64)
-    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)