bug fix

2019-03-11 14:13:32 +01:00 · 2019-03-11 14:13:32 +01:00 · 5cbe0f83f6
parent 4ffda89c42
commit 5cbe0f83f6
1 changed files with 1 additions and 1 deletions
--- 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)