Unverified Commit a2b70758 authored by Prabhat Roy's avatar Prabhat Roy Committed by GitHub
Browse files

Remove unnecessary initialisation (#5507)

parent cef22ea8
......@@ -29,8 +29,7 @@ torch::Tensor GPUDecoder::decode() {
unsigned long videoBytes = 0;
uint8_t* video = nullptr;
at::cuda::CUDAGuard device_guard(device);
auto options = torch::TensorOptions().dtype(torch::kU8).device(torch::kCUDA);
torch::Tensor frame = torch::zeros({0}, options);
torch::Tensor frame;
do {
demuxer.demux(&video, &videoBytes);
decoder.decode(video, videoBytes);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment