o i @sdZddlmZmZmZmZddlmZdeeefde de dee e ffdd Z deeefd eeefd e de fd d Z deeefdeeeefd e dee e ffddZ dS)z0Set of methods to validate encoder architecture.)AnyDictListTuple)sub_factor_to_params configurationblock_idprevious_block_outputreturncCs|d}|durtd||dvr;|ddur td||ddur-td||d }|d }||fS|d kre|d }|durNtd ||d dur[td||}|d<||fStd|)aValidate block arguments. Args: configuration: Architecture configuration. block_id: Block ID. previous_block_output: Previous block output size. Returns: input_size: Block input size. output_size: Block output size. block_typeNz2Block %d in encoder doesn't have a type assigned. ) branchformer conformer linear_sizez:Missing 'linear_size' argument for X-former block (ID: %d)conv_mod_kernel_sizezCMissing 'conv_mod_kernel_size' argument for X-former block (ID: %d) hidden_sizeconv1d output_sizez8Missing 'output_size' argument for Conv1d block (ID: %d) kernel_sizez8Missing 'kernel_size' argument for Conv1d block (ID: %d) input_sizez Block type: %s is not supported.)get ValueError)rrr r rrr]/home/ubuntu/.local/lib/python3.10/site-packages/espnet2/asr_transducer/encoder/validation.pyvalidate_block_argumentss@      rbody_first_confrc Cs4|dd}|d}gd}|dus||vrdS|ddur$d|d<|r6|d d }t|tr5||f}n|d d}t|trE|d }|d kr|rV|d |dd}n&|dur`|dd}|d}t||\} } } | d ksxJd||f| |}d|d<n|d}|dur|}||d<||d <||d<|S)aValidate input block. Args: configuration: Encoder input block configuration. body_first_conf: Encoder first body block configuration. input_size: Encoder input block input size. Return: output_size: Encoder input block output size. vgg_likeFr )r r rNsubsampling_factor conv_size)@rrrr z>Conv2D output size is <1 with input size %d and subsampling %dr)r isinstanceinttupler) rrrrnext_block_typeallowed_next_block_typerr sub_factor_ conv_osizerrrvalidate_input_blockCsJ          r, input_conf body_confcCst||d|}g}t|D]\}}t||d|dkr|n||dd}||qtdt|D]}||dd||dkrMtd|d|fq3||ddfS)aCValidate specified architecture is valid. Args: input_conf: Encoder input block configuration. body_conf: Encoder body blocks configuration. input_size: Encoder input size. Returns: input_block_osize: Encoder input block output size. : Encoder body block output size. rr"zCOutput/Input mismatch between blocks %d and %d in the encoder body.r)r, enumeraterappendrangelenr)r-r.rinput_block_osizecmp_ioib_iorrrvalidate_architectures "  r8N) __doc__typingrrrrespnet2.asr_transducer.utilsrstrr%rr,r8rrrrs<    ;   E