Merge pull request #212 from NZ99/patch-3

Set pin_memory to True in default dataloader config.

Merge pull request #212 from NZ99/patch-3
Set pin_memory to True in default dataloader config.
b05b6d22 · Gustaf Ahdritz · GitHub · 82397498 · 5d6df144 · b05b6d22
Unverified Commit b05b6d22 authored Sep 01, 2022 by Gustaf Ahdritz Committed by GitHub Sep 01, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

openfold/config.py openfold/config.py +1 -0

No files found.
--- a/openfold/config.py
+++ b/openfold/config.py
@@ -359,6 +359,7 @@ config = mlc.ConfigDict(
                "data_loaders": {
                    "batch_size": 1,
                    "num_workers": 8,
+                    "pin_memory": True,
                },
            },
        },