Fix seeding bug

3927776d · Gustaf Ahdritz · 10b6afd6 · 3927776d
Commit 3927776d authored Jul 11, 2022 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

openfold/data/data_modules.py openfold/data/data_modules.py +3 -1

No files found.
--- a/openfold/data/data_modules.py
+++ b/openfold/data/data_modules.py
@@ -630,7 +630,9 @@ class OpenFoldDataModule(pl.LightningDataModule):

            if(self.batch_seed is not None):
                generator = torch.Generator()
-                generator = generator.manual_seed(self.batch_seed + 1)
+                # A standin for the rank, which I don't know how to get
+                pid = os.getpid() % 10000
+                generator = generator.manual_seed(self.batch_seed + pid)
            
            self.train_dataset = OpenFoldDataset(
                datasets=datasets,