diff --git a/torchvision/csrc/io/video/Video.cpp b/torchvision/csrc/io/video/Video.cpp index 9a995c23e6c..c1e19e0b5f5 100644 --- a/torchvision/csrc/io/video/Video.cpp +++ b/torchvision/csrc/io/video/Video.cpp @@ -311,8 +311,8 @@ std::tuple Video::Next() { // currently not supporting other formats (will do soon) out.payload.reset(); - } else if (res == 61) { - LOG(INFO) << "Decoder ran out of frames (error 61)\n"; + } else if (res == ENODATA) { + LOG(INFO) << "Decoder ran out of frames (ENODATA)\n"; } else { LOG(ERROR) << "Decoder failed with ERROR_CODE " << res; }