We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7e2da5d commit 08236beCopy full SHA for 08236be
benchmarks/decoders/generate_readme_data.py
@@ -61,7 +61,8 @@ def main() -> None:
61
62
decoder_dict = {}
63
decoder_dict["torchcodec"] = TorchCodecPublic()
64
- decoder_dict["torchcodec[cuda]"] = TorchCodecPublic(device="cuda")
+ if torch.cuda.is_available():
65
+ decoder_dict["torchcodec[cuda]"] = TorchCodecPublic(device="cuda")
66
decoder_dict["torchvision[video_reader]"] = TorchVision("video_reader")
67
decoder_dict["torchaudio"] = TorchAudioDecoder()
68
@@ -80,7 +81,7 @@ def main() -> None:
80
81
batch_parameters=BatchParameters(batch_size=50, num_threads=10),
82
resize_height=256,
83
resize_width=256,
- resize_device="cuda",
84
+ resize_device="cuda" if torch.cuda.is_available() else "cpu",
85
),
86
)
87
data_for_writing = {
0 commit comments