train.yaml

# config for training
trainer:
  checkpoint: "models"
  optimizer: "adam"
  lr: 1e-3
  momentum: 0.9    #you don't need to change the value,it is not used
  weight_decay: 0  #you don't need to change the value,it is not used
  num_epoches: 300
  clip_norm: False   #you don't need to change the value,it is not used

#You need change the model_path and epoch_done in temp when retrain the model
temp:
  model_path: '/data/kanwang/TasNET/models/TasNET_125_trainloss_-2.8985e+01_valloss_-2.3524e+01.pkl'
  epoch_done: 89

#parameters for generating data
data_generator:
  speech_dir: '/data/kanwang/data/min/'
  train_save_path: '/data/kanwang/TasNET/data/train_input'
  dev_save_path: '/data/kanwang/TasNET/data/dev_input'
  test_save_path: '/data/kanwang/TasNET/data/test_input'
  num_data: 12780  #you don't need to change the value,it is not used
  sr: 8000
  N_L: 40   #N_L is set to the number corresponding to 5ms under the given sampling rate
  len_time: 0.5   #There 2 training stages: first 0.5s and then 4s 
  is_norm: True   #you don't need to change the value,it is not used

#Parameters for data loader, you only need to change the train_path_npz and valid_path_npz for different datasets
dataloader:
  shuffle: true
  batch_size: 128
  drop_last: true
  train_path_npz: '/data/kanwang/TasNET/data/4s_train_input'
  valid_path_npz: '/data/kanwang/TasNET/data/4s_dev_input'
  test_path_npz: '/data/kanwang/TasNET/data/4s_test_input'

#Parameters for test.py 
test:
  test_path_npz: '/data/kanwang/TasNET/data/temp_test'
  file1: '8-1.wav'    #you don't need to change the value,it is not used
  file2: '8-2.wav'    #you don't need to change the value,it is not used
  test_batch_size: 1
  test_model_path: '/data/kanwang/TasNET/models/TasNET_125_trainloss_-2.8985e+01_valloss_-2.3524e+01.pkl'
  test_save_path: '/data/kanwang/TasNET/test/'   #The mix audio and separated audios will be saved in /mix/, /s1/, /s2/ under test_save_path
  test_load_path: '/data/kanwang/data/min/tt/'
  test_len_time: 4   #you don't need to change the value,it is not used