From 392e37021afe40775af5d61f3470f5fe0b10d653 Mon Sep 17 00:00:00 2001 From: OleehyO Date: Tue, 7 Jan 2025 09:50:21 +0000 Subject: [PATCH] Add video path to error message for better debugging --- finetune/datasets/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune/datasets/utils.py b/finetune/datasets/utils.py index 1048d79..a4c747c 100644 --- a/finetune/datasets/utils.py +++ b/finetune/datasets/utils.py @@ -126,7 +126,7 @@ def preprocess_video_with_resize( video_reader = decord.VideoReader(uri=video_path.as_posix(), width=width, height=height) video_num_frames = len(video_reader) if video_num_frames < max_num_frames: - raise ValueError(f"video's frames is less than {max_num_frames}.") + raise ValueError(f"video frame count in {video_path} is less than {max_num_frames}.") indices = list(range(0, video_num_frames, video_num_frames // max_num_frames)) frames = video_reader.get_batch(indices)