class_docstr: Implementation for https://arxiv.org/abs/2006.04558
class_docstr_tok: ["Implementation","for","https",":","\/\/arxiv.org\/abs\/2006.04558"]
code: def add_args(parser):
      parser.add_argument("--dropout", type=float)
      parser.add_argument("--output-frame-dim", type=int)
      parser.add_argument("--speaker-embed-dim", type=int)
      # FFT blocks
      parser.add_argument("--fft-hidden-dim", type=int)
      parser.add_argument("--fft-kernel-size", type=int)
      parser.add_argument("--attention-dropout", type=float)
      parser.add_argument("--encoder-layers", type=int)
      parser.add_argument("--encoder-embed-dim", type=int)
      parser.add_argument("--encoder-attention-heads", type=int)
      parser.add_argument("--decoder-layers", type=int)
      parser.add_argument("--decoder-embed-dim", type=int)
      parser.add_argument("--decoder-attention-heads", type=int)
      # variance predictor
      parser.add_argument("--var-pred-n-bins", type=int)
      parser.add_argument("--var-pred-hidden-dim", type=int)
      parser.add_argument("--var-pred-kernel-size", type=int)
      parser.add_argument("--var-pred-dropout", type=float)
      # postnet
      parser.add_argument("--add-postnet", action="store_true")
      parser.add_argument("--postnet-dropout", type=float)
      parser.add_argument("--postnet-layers", type=int)
      parser.add_argument("--postnet-conv-dim", type=int)
      parser.add_argument("--postnet-conv-kernel-size", type=int)
  
imports: import logging
  import torch
  from torch import nn
  from fairseq import utils
  from fairseq.data.data_utils import lengths_to_padding_mask
  from fairseq.models import FairseqEncoder, FairseqEncoderModel, register_model, register_model_architecture
  from fairseq.models.text_to_speech.hub_interface import TTSHubInterface
  from fairseq.models.text_to_speech.tacotron2 import Postnet
  from fairseq.modules import FairseqDropout, LayerNorm, MultiheadAttention, PositionalEmbedding
label_desc: Use image node_id 3 for calling the FastSpeech2Model obj's underlying member method code with example usage: obj.add_args(parser) without return types