Skip to content

GitLab

Commit dde7e6c4 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Fix another data module bug

parent 33519974
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
openfold/data/data_modules.py
View file @ dde7e6c4
...@@ -213,7 +213,6 @@ class OpenFoldDataset(torch.utils.data.IterableDataset): ...@@ -213,7 +213,6 @@ class OpenFoldDataset(torch.utils.data.IterableDataset):
self.samplers = [ self.samplers = [
looped_sequence(RandomSampler(d)) for d in datasets looped_sequence(RandomSampler(d)) for d in datasets
] ]
self.batch_size = batch_size
self.epoch_len = epoch_len self.epoch_len = epoch_len
self.distr = torch.distributions.categorical.Categorical( self.distr = torch.distributions.categorical.Categorical(
......
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