diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml index a414570d..3cb31e56 100644 --- a/.github/workflows/benchmarks.yaml +++ b/.github/workflows/benchmarks.yaml @@ -32,10 +32,9 @@ jobs: run: python -m pip install --upgrade pip - name: Install dependencies and FFmpeg run: | - # TODO: torchvision and torchaudio shouldn't be needed. They were only added - # to silence an error as seen in https://github.com/pytorch/torchcodec/issues/203 + conda install ffmpeg pkg-config -c conda-forge python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu - conda install "ffmpeg=7.0.1" pkg-config -c conda-forge + python -m pip install decord matplotlib pandas numpy ffmpeg -version - name: Build and install torchcodec run: | @@ -46,4 +45,3 @@ jobs: - name: TEST README data geeneration benchmark run: | python benchmarks/decoders/generate_readme_data.py --test_run - diff --git a/benchmarks/decoders/generate_readme_data.py b/benchmarks/decoders/generate_readme_data.py index 57beda50..f8e24538 100644 --- a/benchmarks/decoders/generate_readme_data.py +++ b/benchmarks/decoders/generate_readme_data.py @@ -58,7 +58,7 @@ def main() -> None: patterns = ["mandelbrot"] fpses = [30] gop_sizes = [20] - durations = [10] # if this goes too low, we hit EOF errors in some decoders + durations = [10] # if this goes too low, we hit EOF errors in some decoders pix_fmts = ["yuv420p"] ffmpeg_path = "ffmpeg" min_runtime_seconds = 1