log_dir: "Models/VCTK"
first_stage_path: "first_stage.pth"
save_freq: 2
log_interval: 10
device: "cuda"
multigpu: false
epochs_1st: 150 # number of epochs for first stage training
epochs_2nd: 100 # number of peochs for second stage training
batch_size: 32
pretrained_model: ""
second_stage_load_pretrained: false # set to true if the pre-trained model is for 2nd stage
load_only_params: false # set to true if do not want to load epoch numbers and optimizer parameters

train_data: "Data/train_list.txt"
val_data: "Data/val_list.txt"

F0_path: "Utils/JDC/bst.t7"
ASR_config: "Utils/ASR/config.yml"
ASR_path: "Utils/ASR/epoch_00080.pth"

preprocess_params:
  sr: 24000
  spect_params:
    n_fft: 2048
    win_length: 1200
    hop_length: 300

model_params:
  hidden_dim: 512
  n_token: 178
  style_dim: 128
  n_layer: 3
  dim_in: 64
  max_conv_dim: 512
  n_mels: 80
  dropout: 0.2
  n_domain: 108

loss_params:
    lambda_mel: 5. # mel reconstruction loss (1st & 2nd stage)
    lambda_adv: 1. # adversarial loss (1st & 2nd stage)
    lambda_reg: 1. # adversarial regularization loss (1st & 2nd stage)
    lambda_fm: 0.1 # feature matching loss (1st & 2nd stage)
        
    lambda_mono: 1. # monotonic alignment loss (1st stage, TMA)
    lambda_s2s: 1. # sequence-to-sequence loss (1st stage, TMA)
    lambda_sty: 1. # style reconstruction loss (1st stage)
    
    TMA_epoch: 10 # TMA starting epoch (1st stage)
    VC_epoch: -5 # VC starting epoch (1st stage), only fine-tune for VC for 5 epochs
    TMA_CEloss: false # see https://github.com/yl4579/StyleTTS/issues/7

    lambda_feat: 1. # feature reconstruction loss (1st stage)
    lambda_pim: 1. # phoneme information maximazation loss (2nd stage)
    lambda_cyc: 1. # cycle consistency loss (2nd stage)


optimizer_params:
  lr: 0.0001