o
    
j                     @   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)link_arguments)selfparser r
   G/home/kuhnn/.local/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)levelTF
max_epochs)trainer_defaults)loggingbasicConfigINFOtorchbackendscudamatmul
allow_tf32cudnndeterministicr   r   r   )_clir
   r
   r   main   s   



r   __main__)r   r   lightning.pytorch.clir   vits.datasetr   vits.lightningr   	getLogger__package___LOGGERr   r   r   r
   r
   r
   r   <module>   s    

