| 346 | |
| 347 | class FastspeechDecoder(FFTBlocks): |
| 348 | def __init__(self, hidden_size=None, num_layers=None, kernel_size=None, num_heads=None): |
| 349 | num_heads = hparams['num_heads'] if num_heads is None else num_heads |
| 350 | hidden_size = hparams['hidden_size'] if hidden_size is None else hidden_size |
| 351 | kernel_size = hparams['dec_ffn_kernel_size'] if kernel_size is None else kernel_size |
| 352 | num_layers = hparams['dec_layers'] if num_layers is None else num_layers |
| 353 | super().__init__(hidden_size, num_layers, kernel_size, num_heads=num_heads) |
| 354 | |