o
    Wli                     @   sl   d dl Z d dlZd dlmZ ddlmZ ddlmZ e e	Z
G dd deZdd	 Zed
kr4e  dS dS )    N)LightningCLI   )VitsDataModule)	VitsModelc                   @   s   e Zd Zdd ZdS )VitsLightningCLIc                 C   sd   | dd | dd | dd | dd | d	d
 | dd | dd | dd d S )Nzdata.batch_sizezmodel.batch_sizezdata.num_symbolszmodel.num_symbolszmodel.num_speakerszdata.num_speakerszmodel.sample_ratezdata.sample_ratezmodel.filter_lengthzdata.filter_lengthzmodel.hop_lengthzdata.hop_lengthzmodel.win_lengthzdata.win_lengthzmodel.segment_sizezdata.segment_size)Zlink_arguments)selfparser r	   N/home/kim/smarthome/.venv/lib/python3.10/site-packages/piper/train/__main__.pyadd_arguments_to_parser   s   z(VitsLightningCLI.add_arguments_to_parserN)__name__
__module____qualname__r   r	   r	   r	   r
   r      s    r   c                  C   sD   t jt jd dtjjj_dtjj_dtjj_	t
ttddid} d S )N)levelTFZ
max_epochs)Ztrainer_defaults)loggingbasicConfigINFOtorchbackendscudamatmulZ
allow_tf32ZcudnnZdeterministicr   r   r   )Z_clir	   r	   r
   main   s   



r   __main__)r   r   Zlightning.pytorch.clir   Zvits.datasetr   Zvits.lightningr   	getLogger__package__Z_LOGGERr   r   r   r	   r	   r	   r
   <module>   s    

