From 8e39ce128b097963fc250e0e0bfe253dacae9c2b Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 18 Apr 2018 15:25:35 -0700 Subject: [PATCH] config change CUDA_VISIBLE_DEVICES=0 python train.py --config_path config.json --restore_path /data/shared/erogol_models/April-16-2018_01:29PM-41e942e/checkpoint_75576.pth.tar --- config.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.json b/config.json index 6a28da75..726dbf83 100644 --- a/config.json +++ b/config.json @@ -10,7 +10,7 @@ "embedding_size": 256, "text_cleaner": "english_cleaners", - "epochs": 200, + "epochs": 500, "lr": 0.002, "warmup_steps": 4000, "batch_size": 32,