o
    oi                     @   sL   d dl mZ d dlZd dlmZ d dlmZmZ G dd deZdd Z	dS )	    )import_moduleN)logger)DfParamsconfigc                   @   s"   e Zd Zdd ZdefddZdS )ModelParamsc                 C   s,   t dddd| _ttd| j d | _d S )NMODELdeepfilternet3traindefaultsectiondf.r   )r   _ModelParams__modelgetattrr   _ModelParams__params)self r   F/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/df/model.py__init__
   s   zModelParams.__init__attrc                 C   s   t | j|S )N)r   r   )r   r   r   r   r   __getattr__   s   zModelParams.__getattr__N)__name__
__module____qualname__r   strr   r   r   r   r   r   	   s    r   c                  O   sN   t dddd}td| d ttd| d| i |}|jtjd	 |S )
z-Initialize the model specified in the config.r   r   r	   r
   zInitializing model ``r   
init_model)memory_format)r   r   infor   r   totorchchannels_last)argskwargsmodelr   r   r   r      s
   r   )
	importlibr   r    logurur   	df.configr   r   r   r   r   r   r   r   <module>   s    	