Просмотр исходного кода

Optimize vqgan finetune config

Lengyue 1 год назад
Родитель
Сommit
d1f92c5669
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      fish_speech/configs/vqgan_finetune.yaml

+ 1 - 1
fish_speech/configs/vqgan_finetune.yaml

@@ -117,7 +117,7 @@ model:
     lr_lambda:
       _target_: fish_speech.scheduler.get_cosine_schedule_with_warmup_lr_lambda
       _partial_: true
-      num_warmup_steps: 100
+      num_warmup_steps: 0
       num_training_steps: ${trainer.max_steps}
       final_lr_ratio: 0