Skip to content

GitLab

Unverified Commit b05b6d22 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz Committed by GitHub
Browse files

Merge pull request #212 from NZ99/patch-3 

Set pin_memory to True in default dataloader config.
parents 82397498 5d6df144
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
openfold/config.py
View file @ b05b6d22
...@@ -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