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.
2 parents a269432 + a2b7075 commit 364da8fCopy full SHA for 364da8f
torchvision/csrc/io/decoder/gpu/gpu_decoder.cpp
@@ -29,8 +29,7 @@ torch::Tensor GPUDecoder::decode() {
29
unsigned long videoBytes = 0;
30
uint8_t* video = nullptr;
31
at::cuda::CUDAGuard device_guard(device);
32
- auto options = torch::TensorOptions().dtype(torch::kU8).device(torch::kCUDA);
33
- torch::Tensor frame = torch::zeros({0}, options);
+ torch::Tensor frame;
34
do {
35
demuxer.demux(&video, &videoBytes);
36
decoder.decode(video, videoBytes);
0 commit comments