fix a bug related to audio_end_pts (#1431)

pytorch · Oct 8, 2019 · 1258bb7 · 1258bb7
1 parent ef0ffb8
commit 1258bb7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/datasets/video_utils.py b/torchvision/datasets/video_utils.py
@@ -308,7 +308,7 @@ def get_clip(self, idx):
                     math.floor,
                 )
                 audio_end_pts = pts_convert(
-                    video_start_pts,
+                    video_end_pts,
                     info["video_timebase"],
                     info["audio_timebase"],
                     math.ceil,