Set pin_memory to True in default dataloader config.

5d6df144 · Niccolò Zanichelli · GitHub · 9dd9cea4 · 5d6df144
Unverified Commit 5d6df144 authored Aug 31, 2022 by Niccolò Zanichelli Committed by GitHub Aug 31, 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,
                },
            },
        },