attempt to fix 103, 108

wjtscn · Feb 7, 2020 · 8d7f9df · 8d7f9df
1 parent 8253256
commit 8d7f9df
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/datasets/imgnetvid.py b/datasets/imgnetvid.py
@@ -191,6 +191,7 @@ def __getitem__(self, idx):
                 imgs = list()
                 lbls = list()
                 window_sample_ids = self._windows[self.sample_ids[idx]]
+                window_sample_ids = window_sample_ids[:self._window_size]
 
                 # go through the sample ids for the window
                 for sid in window_sample_ids: