Fix yet ANOTHER OBO error in multi_frame_dataset
This commit is contained in:
parent
0cf52ef52c
commit
6be6c92e5d
|
@ -19,10 +19,9 @@ class MultiFrameDataset(BaseUnsupervisedImageDataset):
|
|||
frames_needed = self.num_frames
|
||||
search_idx = chunk_index
|
||||
while frames_needed > 0:
|
||||
if source_name in self.chunk_name(search_idx):
|
||||
frames_needed -= 1
|
||||
search_idx -= 1
|
||||
else:
|
||||
if source_name not in self.chunk_name(search_idx):
|
||||
search_idx += 1
|
||||
break
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user