Fix drop_last=True in the data loader for video evaluation

zlu8899 · Feb 19, 2024 · 5243a08 · 5243a08
1 parent 2dfdfac
commit 5243a08
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)