Skip to content

Commit 364da8f

Browse files
authored
Merge branch 'main' into models/efficientnet_v2
2 parents a269432 + a2b7075 commit 364da8f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

torchvision/csrc/io/decoder/gpu/gpu_decoder.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,7 @@ torch::Tensor GPUDecoder::decode() {
2929
unsigned long videoBytes = 0;
3030
uint8_t* video = nullptr;
3131
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);
32+
torch::Tensor frame;
3433
do {
3534
demuxer.demux(&video, &videoBytes);
3635
decoder.decode(video, videoBytes);

0 commit comments

Comments
 (0)