Skip to content

GitLab

Unverified Commit 5d6df144 authored by Niccolò Zanichelli's avatar Niccolò Zanichelli Committed by GitHub
Browse files

Set pin_memory to True in default dataloader config.

parent 9dd9cea4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
openfold/config.py
View file @ 5d6df144
...@@ -359,6 +359,7 @@ config = mlc.ConfigDict( ...@@ -359,6 +359,7 @@ config = mlc.ConfigDict(
"data_loaders": { "data_loaders": {
"batch_size": 1, "batch_size": 1,
"num_workers": 8, "num_workers": 8,
"pin_memory": True,
}, },
}, },
}, },
......
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