egs/proposed/bin/train.py

# Copyright 2024 LY Corporation

# LY Corporation licenses this file to you under the Apache License,
# version 2.0 (the "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at:

#   https://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import hydra
from omegaconf import DictConfig
from promptttspp.trainers.tts import TTSTrainer


@hydra.main(version_base=None, config_path="conf/", config_name="train")
def main(cfg: DictConfig):
    trainer = TTSTrainer(cfg)
    trainer.run()


if __name__ == "__main__":
    main()