Fix tolerance for delta_timestamps (#84)
Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user