Respair
/

stts_test_ckpt

Respair commited on 17 days ago

Commit

83b9bee

verified ·

1 Parent(s): ae278e3

Update StyleTTS_Accelerate/Configs/config.yml

Files changed (1) hide show

StyleTTS_Accelerate/Configs/config.yml CHANGED Viewed

@@ -6,7 +6,7 @@ device: "cuda"
 multigpu: false
 epochs_1st: 200 # number of epochs for first stage training
 epochs_2nd: 100 # number of peochs for second stage training
-batch_size: 24
 pretrained_model: "/kaggle/working/StyleTTS_Accelerate/Models/Anispeech_with_DIFF_kaggle/epoch_2nd_00004.pth"
 second_stage_load_pretrained: true # set to true if the pre-trained model is for 2nd stage
 load_only_params: true # set to true if do not want to load epoch numbers and optimizer parameters

 multigpu: false
 epochs_1st: 200 # number of epochs for first stage training
 epochs_2nd: 100 # number of peochs for second stage training
+batch_size: 22
 pretrained_model: "/kaggle/working/StyleTTS_Accelerate/Models/Anispeech_with_DIFF_kaggle/epoch_2nd_00004.pth"
 second_stage_load_pretrained: true # set to true if the pre-trained model is for 2nd stage
 load_only_params: true # set to true if do not want to load epoch numbers and optimizer parameters