o }oi@sjddlmZmZddlmZeGdddZeGdddZeGdddZeGd d d eZd S) ) dataclassfieldGraphModuleConfigc@s&eZdZUdZeed<dZeed<dS)AlignerCTCConfigprob_suppress_indexg?prob_suppress_valueN)__name__ __module__ __qualname__rint__annotations__r floatrrf/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/asr/models/configs/aligner_config.pyr rc@s&eZdZUdZeed<dZeed<dS)AlignerRNNTConfigrpredictor_window_sizepredictor_step_sizeN)r r r rr rrrrrrrrrc@sjeZdZUdZeed<dZeed<dZeed<dZ e ed<e d d d Z e ed <e d d d Zeed<dS)AlignerWrapperModelConfigforcedalignment_typeT word_outputF cpu_decodingrdecode_batch_sizecCtSN)rrrrr&z"AlignerWrapperModelConfig.default_factoryctc_cfgcCrr)rrrrrr'r rnnt_cfgN)r r r rstrrrboolrrr rr#rr$rrrrrr s    rc@s$eZdZUedddZeed<dS)K2AlignerWrapperModelConfigcCrrrrrrrr,r z$K2AlignerWrapperModelConfig.r!decoder_module_cfgN)r r r rr(rrrrrrr'*s r'N) dataclassesrr%nemo.collections.asr.parts.k2.classesrrrrr'rrrrs