fix a bug related to audio_end_pts (#1431)

1258bb7b · Zhicheng Yan · Francisco Massa · ef0ffb80 · 1258bb7b
Commit 1258bb7b authored Oct 08, 2019 by Zhicheng Yan Committed by Francisco Massa Oct 08, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/datasets/video_utils.py torchvision/datasets/video_utils.py +1 -1

No files found.
--- a/torchvision/datasets/video_utils.py
+++ b/torchvision/datasets/video_utils.py
@@ -308,7 +308,7 @@ class VideoClips(object):
                    math.floor,
                )
                audio_end_pts = pts_convert(
-                    video_start_pts,
+                    video_end_pts,
                    info["video_timebase"],
                    info["audio_timebase"],
                    math.ceil,