Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix bug for VITS resuming training #108

Merged
merged 2 commits into from
Jan 17, 2024

Conversation

lmxue
Copy link
Collaborator

@lmxue lmxue commented Jan 15, 2024

✨ Description

  • Fix the bug for VITS resuming training
  • Add support for VITS resuming training and fine-tuning.
  • Have tested on VITS, including training from scratch, resuming training and fine-tuning, and VALL-E, including stage 1 training and stage 2 training.

🚧 Related Issues

Issue #94

πŸ‘¨β€πŸ’» Changes Proposed

  • Add a user-friendly interface in egs/tts/VITS/run.sh to use the arguments related to resuming or fine-tuning VITS model.
  • Add support in models/tts/base/tts_trainer.py for fine-tuning VITS model.
  • Add usage instructions in egs/tts/VITS/README.md about training VITS from scratch, resuming, or finetuning.

πŸ§‘β€πŸ€β€πŸ§‘ Who Can Review?

@viewfinder-annn @RMSnow @VocodexElysium

πŸ›  TODO

No TODO.

βœ… Checklist

  • Code has been reviewed
  • Code complies with the project's code standards and best practices
  • Code has passed all tests
  • Code does not affect the normal use of existing features
  • Code has been commented properly
  • Documentation has been updated (if applicable)
  • Demo/checkpoint has been attached (if applicable)

Copy link
Collaborator

@RMSnow RMSnow left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@lmxue lmxue mentioned this pull request Jan 16, 2024
@lmxue lmxue merged commit 4125584 into open-mmlab:main Jan 17, 2024
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants