mohan32189/Sanskrit-TTS
0
1B
2 �)c� � @ sb d dl Z d dlmZ d dlmZmZ d dlmZ d dlZG dd� d�Z dd� Z3d d4� Z d d � Z dS )
� N)�loads)�load� FloatTensor)�float32c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d5� Zd d � Zd
d� Z dd� Z6dd� Z dS )�HParamsc K s8 x2|� � D ]&\}}t|�tkr(tf |�}|| |< q7W d S )N)�items�type�dictr )�self�kwargs�k�v� r �!e:\Projects\Sanskrit-TTS\utils.py�__init__ s 8zHParams.__init__c C s9 | j �� S )N)�__dict__�keys)r10 r r r r s z HParams.keysc C s11 | j �� S )N)r r )r12 r r r r s z
HParams.itemsc C s13 | j �� S )N)r �values)r14 r r r r s zHParams.valuesc C s15 t | j�S )N)�lenr )r16 r r r �__len__ s zHParams.__len__c C s17 t | |�S )N)�getattr)r18 �keyr r r � __getitem__ s zHParams.__getitem__c C s t | ||�S )N)�setattr)r19 r �valuer r r � __setitem__ s zHParams.__setitem__c C s20 || j kS )N)r )r21 r r r r � __contains__! s zHParams.__contains__c C s22 | j �� S )N)r �__repr__)r23 r r r r $ s zHParams.__repr__N) �__name__�24__module__� __qualname__r r r r r r r r r r r r r r s r c C s� t | dd�}|d }|d }t|d�r2|j�� }n|�� }i }xH|�� D ]<\}}y|| ||< W qH t�d| � |||< Y qHX qHW t|d�r�|j�|� n25|�|� t�d�| |�� d S )N�cpu)� map_location� iteration�model�modulez%s is not in the checkpointz%Loaded checkpoint '{}' (iteration {})) r �hasattrr% �26state_dictr �logging�info�load_state_dict�format) Zcheckpoint_pathr$ Zcheckpoint_dictr# Zsaved_state_dictr' Znew_state_dictr r
r r r �load_checkpoint( s& 27 282930r, c C s4 t | d��}|�� }W d Q R X t|�}tf |�}|S )N�r)�open�readr r )Z config_path�f�data�configZhparamsr r r �get_hparams_from_file@ s31 32r3 c C s" t j| |dd�\}}t|�t��S )NT)�sr�mono)�librosar r �astyper )� full_pathZtarget_sampling_rateZaudioZ
sampling_rater r r �load_audio_to_torchI s r9 )
r( �jsonr �torchr r �numpyr r6 r r, r3 r9 r r r r �<module> s