Merge pull request facebookresearch#11 from bryant1410/patch-4

Fix `drop_last=True` in the data loader for video evaluation
zlu8899 · Feb 19, 2024 · 65fff29 · 65fff29
2 parents ef10bd3 + 5243a08
commit 65fff29
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/datasets/data_manager.py b/src/datasets/data_manager.py
@@ -85,6 +85,7 @@ def init_data(
             num_workers=num_workers,
             world_size=world_size,
             rank=rank,
+            drop_last=drop_last,
             log_dir=log_dir)
 
     return (data_loader, dist_sampler)