TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
data_function.TextMelCollate Class Reference

Public Member Functions

def __init__ (self, n_frames_per_step)
 
def __call__ (self, batch)
 

Public Attributes

 n_frames_per_step
 

Detailed Description

Zero-pads model inputs and targets based on number of frames per setep

Constructor & Destructor Documentation

◆ __init__()

def data_function.TextMelCollate.__init__ (   self,
  n_frames_per_step 
)

Member Function Documentation

◆ __call__()

def data_function.TextMelCollate.__call__ (   self,
  batch 
)
Collate's training batch from normalized text and mel-spectrogram
PARAMS
------
batch: [text_normalized, mel_normalized]

Member Data Documentation

◆ n_frames_per_step

data_function.TextMelCollate.n_frames_per_step

The documentation for this class was generated from the following file: