Add test.py for gpu decoder

Works with SliceSampler WIP Add video_id_to_path as a meta_data plot frame
2024-03-13 16:27:44 +00:00
parent 75cc10198f
commit d32a279435
3 changed files with 597 additions and 15 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -51,6 +51,7 @@ torchvision = "^0.17.1"
 h5py = "^3.10.0"
 dm-control = "1.0.14"
 huggingface-hub = {extras = ["hf-transfer"], version = "^0.21.4"}
+torchaudio = "^2.2.1"


 [tool.poetry.group.dev.dependencies]