avois OS memory error with num_workers > 1

This commit is contained in:
ferdinand.mom 2024-12-02 18:33:35 +00:00
parent 75939867d9
commit 86a0fc5e3d

View File

@ -29,7 +29,7 @@
}, },
"dataset": { "dataset": {
"name": "roneneldan/TinyStories", "name": "roneneldan/TinyStories",
"num_workers": 1, "num_workers": 0,
"num_proc": 4 "num_proc": 4
}, },
"checkpoint": { "checkpoint": {