diff --git a/template/base_config.json b/template/base_config.json index 7b57675..342f7ba 100644 --- a/template/base_config.json +++ b/template/base_config.json @@ -29,7 +29,7 @@ }, "dataset": { "name": "roneneldan/TinyStories", - "num_workers": 4, + "num_workers": 1, "num_proc": 4 }, "checkpoint": {