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() { ...@@ -29,8 +29,7 @@ torch::Tensor GPUDecoder::decode() {
unsigned long videoBytes = 0; unsigned long videoBytes = 0;
uint8_t* video = nullptr; uint8_t* video = nullptr;
at::cuda::CUDAGuard device_guard(device); at::cuda::CUDAGuard device_guard(device);
auto options = torch::TensorOptions().dtype(torch::kU8).device(torch::kCUDA); torch::Tensor frame;
torch::Tensor frame = torch::zeros({0}, options);
do { do {
demuxer.demux(&video, &videoBytes); demuxer.demux(&video, &videoBytes);
decoder.decode(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