From 410e182902a92a0d73928e764149d5444ec98a88 Mon Sep 17 00:00:00 2001 From: Gustaf Ahdritz Date: Wed, 9 Aug 2023 00:20:32 -0400 Subject: [PATCH] Fix another seeding bug --- openfold/data/input_pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/openfold/data/input_pipeline.py b/openfold/data/input_pipeline.py index 779fed77..64ba4c6e 100644 --- a/openfold/data/input_pipeline.py +++ b/openfold/data/input_pipeline.py @@ -14,6 +14,7 @@ # limitations under the License. from functools import partial +import random import torch @@ -153,7 +154,7 @@ def ensembled_transform_fns(common_cfg, mode_cfg, ensemble_seed): def process_tensors_from_config(tensors, common_cfg, mode_cfg): """Based on the config, apply filters and transformations to the data.""" - ensemble_seed = torch.Generator().seed() + ensemble_seed = random.randint(0, 2147483647) def wrap_ensemble_fn(data, i): """Function to be mapped over the ensemble dimension."""