diff --git a/default_config.yml b/default_config.yml index ab5bebc..99d7601 100644 --- a/default_config.yml +++ b/default_config.yml @@ -22,7 +22,7 @@ preprocess_text: bert_gen: config_path: "config.json" - num_processes: 4 + num_processes: 2 device: "cuda" use_multi_device: false