dump dataset, unset distributed config

rwth-i6 · May 10, 2024 · fa21d00 · fa21d00
1 parent a7e47db
commit fa21d00
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/dump-dataset.py b/tools/dump-dataset.py
@@ -236,6 +236,8 @@ def init(config_str, config_dataset, verbosity):
     config = rnn.config
     config.set("log", None)
     config.set("log_verbosity", verbosity)
+    config.set("torch_distributed", None)
+    config.set("use_horovod", None)
     if dataset_dict:
         assert not config_dataset
         dataset = init_dataset(dataset_dict)