Skip to content

GitLab

Commit 809a9861 authored by Kolja Stahl's avatar Kolja Stahl
Browse files

fix collator and cropping

parent e9d2d893
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
openfold/data/data_modules.py
View file @ 809a9861
...@@ -505,7 +505,7 @@ class OpenFoldDataModule(pl.LightningDataModule): ...@@ -505,7 +505,7 @@ class OpenFoldDataModule(pl.LightningDataModule):
else: else:
raise ValueError("Invalid stage") raise ValueError("Invalid stage")
batch_collator = OpenFoldBatchCollator(self.config, stage) batch_collator = OpenFoldBatchCollator(self.config, generator, stage)
dl = OpenFoldDataLoader( dl = OpenFoldDataLoader(
dataset, dataset,
......
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