From 5a9f8fe38f2af6247d9117bcb1a6e071dfa2fd58 Mon Sep 17 00:00:00 2001
From: Eric Alcaide <ericalcaide1@gmail.com>
Date: Tue, 26 Dec 2023 04:01:34 +0100
Subject: [PATCH] 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: https://github.com/dptech-corp/Uni-Core/pull/39
---
 unimol/unimol/tasks/unimol.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/unimol/unimol/tasks/unimol.py b/unimol/unimol/tasks/unimol.py
index f6b5242..ca787a8 100644
--- 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
+