Skip to content

GitLab

"deploy/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "21b795e8311975874b2369e886e805c32d1c7c68"
Unverified Commit 451642c8 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz Committed by GitHub
Browse files

Fix seeding bug

parent 1d878a12
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
openfold/data/data_transforms.py
View file @ 451642c8
...@@ -187,6 +187,8 @@ def sample_msa(protein, max_seq, keep_extra, seed=None): ...@@ -187,6 +187,8 @@ def sample_msa(protein, max_seq, keep_extra, seed=None):
g = torch.Generator(device=protein["msa"].device) g = torch.Generator(device=protein["msa"].device)
if seed is not None: if seed is not None:
g.manual_seed(seed) g.manual_seed(seed)
else:
g.seed()
shuffled = torch.randperm(num_seq - 1, generator=g) + 1 shuffled = torch.randperm(num_seq - 1, generator=g) + 1
index_order = torch.cat( index_order = torch.cat(
(torch.tensor([0], device=shuffled.device), shuffled), (torch.tensor([0], device=shuffled.device), shuffled),
......
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