Skip to content

GitLab

Commit 965961af authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Fix generator bug

parent 3de3bcb9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
openfold/data/data_modules.py
View file @ 965961af
...@@ -631,6 +631,7 @@ class OpenFoldDataModule(pl.LightningDataModule): ...@@ -631,6 +631,7 @@ class OpenFoldDataModule(pl.LightningDataModule):
self.train_chain_data_cache_path, self.train_chain_data_cache_path,
] ]
generator = None
if(self.batch_seed is not None): if(self.batch_seed is not None):
generator = torch.Generator() generator = torch.Generator()
generator = generator.manual_seed(self.batch_seed + 1) generator = generator.manual_seed(self.batch_seed + 1)
......
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