search

pytorch / torchcodec

PyTorch video decoding
BSD 3-Clause "New" or "Revised" License
87 stars 10 forks source link

Set thread_count to a value set by FFMPEG in single-video benchmarks and to 1 in concurrent benchmarks #406

Closed ahmadsharif1 closed 1 week ago

ahmadsharif1 commented 1 week ago
  1. Set ffmpeg thread count to 0 for single video benchmarks. This should saturate the system.
  2. Set ffmpeg thread count to 1 for concurrent benchmarks. This should saturate the system because we have concurrency at the layer above the decoder.
  3. Call concurrent benchmarks "concurrent" instead of "dataloader" as they don't technically use the pytorch dataloader.
  4. Print the benchmark that's about to be run on the screen. This is only about 10 lines of output and makes it clear which benchmark takes a long time to finish.