Skip to content

Commit

Permalink
Fix another seeding bug
Browse files Browse the repository at this point in the history
  • Loading branch information
gahdritz authored Aug 9, 2023
1 parent 451642c commit 410e182
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion openfold/data/input_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
# limitations under the License.

from functools import partial
import random

import torch

Expand Down Expand Up @@ -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."""
Expand Down

0 comments on commit 410e182

Please sign in to comment.