From 5cbe0f83f6835d260f5e6706e27e76101c4b8472 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Mon, 11 Mar 2019 14:13:32 +0100 Subject: [PATCH] bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 5aed4073..b6415e6b 100644 --- a/train.py +++ b/train.py @@ -376,7 +376,7 @@ def main(args): init_distributed(args.rank, num_gpus, args.group_id, c.distributed["backend"], c.distributed["url"]) num_chars = len(phonemes) if c.use_phonemes else len(symbols) - model = MyModel(num_chars=num_chars, r=1) + model = MyModel(num_chars=num_chars, r=c.r) print(" | > Num output units : {}".format(ap.num_freq), flush=True)