about train error

[yyh565655555]

RuntimeError: Input tensor at index 1 has invalid shape [13, 432, 154], but expected [13, 432, 183]

[AI4VTV]

@yyh565655555 comment out line number 90 in train.py which uses nn.DataParallel() for LJSpeech. it will solve the problem.

Also comment out model.Module.state_dict() to model.state_dict() in train.py