Disable batch shuffling. Only per sample (#196)

disable batch shuffling. only per sample Currently batch size will affect the order of samples, therefore can't adjust the effective batch_size (num_gpus * batch_size * update_freq) and keep the same results. See for context: dptech-corp/Uni-Core#39
deepmodeling · Dec 26, 2023 · 5a9f8fe · 5a9f8fe
1 parent d4d5590
commit 5a9f8fe
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/unimol/unimol/tasks/unimol.py b/unimol/unimol/tasks/unimol.py
@@ -235,3 +235,7 @@ def build_model(self, args):
 
         model = models.build_model(args, self)
         return model
+
+    def disable_shuffling(self) -> bool:
+        return True
+