Spaces:
Paused
Paused
Update update_config.py
Browse files- update_config.py +5 -1
update_config.py
CHANGED
@@ -10,10 +10,14 @@ from pathlib import Path
|
|
10 |
def main(config_file="configs/44k/config.json"):
|
11 |
config_path = Path(config_file)
|
12 |
data = json.loads(config_path.read_text())
|
|
|
|
|
13 |
data['train']['num_workers'] = 0
|
14 |
data['train']['persistent_workers'] = False
|
|
|
|
|
|
|
15 |
config_path.write_text(json.dumps(data, indent=2, sort_keys=False))
|
16 |
|
17 |
-
|
18 |
if __name__ == "__main__":
|
19 |
main()
|
|
|
10 |
def main(config_file="configs/44k/config.json"):
|
11 |
config_path = Path(config_file)
|
12 |
data = json.loads(config_path.read_text())
|
13 |
+
data['train']['batch_size'] = 16
|
14 |
+
data['train']['eval_interval'] = 800
|
15 |
data['train']['num_workers'] = 0
|
16 |
data['train']['persistent_workers'] = False
|
17 |
+
data['train']['push_to_hub'] = True
|
18 |
+
data['train']['repo_id'] = tuple(data['spk'])[0]
|
19 |
+
data['train']['private'] = True
|
20 |
config_path.write_text(json.dumps(data, indent=2, sort_keys=False))
|
21 |
|
|
|
22 |
if __name__ == "__main__":
|
23 |
main()
|