Change seeding

8184691b · Gustaf Ahdritz · 47fccce1 · 8184691b
Commit 8184691b authored Jul 20, 2022 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

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

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

            if(self.batch_seed is not None):
                generator = torch.Generator()
-                # 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)
+                generator = generator.manual_seed(self.batch_seed + 1)
            
            self.train_dataset = OpenFoldDataset(
                datasets=datasets,