diff --git a/config.json b/config.json index f362de41..ace6a721 100644 --- a/config.json +++ b/config.json @@ -10,20 +10,20 @@ "embedding_size": 256, "text_cleaner": "english_cleaners", - "epochs": 2000, - "lr": 0.003, + "epochs": 200, + "lr": 0.002, "warmup_steps": 4000, - "batch_size": 128, + "batch_size": 32, "eval_batch_size":32, "r": 5, "griffin_lim_iters": 60, "power": 1.5, - "num_loader_workers": 12, + "num_loader_workers": 8, "checkpoint": true, - "save_step": 94, + "save_step": 376, "data_path": "/run/shm/erogol/LJSpeech-1.0", "min_seq_len": 0, "output_path": "/data/shared/erogol_models/"