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()
|