Fix tolerance for delta_timestamps (#84)

Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
Alexander Soare
2024-04-18 18:48:22 +01:00
committed by GitHub
parent 7ad1909641
commit 8d980940a2
4 changed files with 27 additions and 14 deletions

View File

@@ -67,6 +67,7 @@ class AlohaDataset(torch.utils.data.Dataset):
item,
self.hf_dataset,
self.delta_timestamps,
tol=1 / self.fps - 1e-4, # 1e-4 to account for possible numerical error
)
# convert images from channel last (PIL) to channel first (pytorch)