Skip to content

GitLab

Commit 8184691b authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Change seeding

parent 47fccce1
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
openfold/data/data_modules.py
View file @ 8184691b
...@@ -630,9 +630,7 @@ class OpenFoldDataModule(pl.LightningDataModule): ...@@ -630,9 +630,7 @@ class OpenFoldDataModule(pl.LightningDataModule):
if(self.batch_seed is not None): if(self.batch_seed is not None):
generator = torch.Generator() generator = torch.Generator()
# A standin for the rank, which I don't know how to get generator = generator.manual_seed(self.batch_seed + 1)
pid = os.getpid() % 10000
generator = generator.manual_seed(self.batch_seed + pid)
self.train_dataset = OpenFoldDataset( self.train_dataset = OpenFoldDataset(
datasets=datasets, datasets=datasets,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment