o i@s dZddZddZddZdS)zConformer common arguments.cCsf|jddtgddd|jddtdd |jd d d td d |jddtdd |jddtdd |S)z(Define common arguments for RNN encoder.z--etypeblstmp)lstmblstmlstmprvgglstmp vggblstmpvgglstmvggblstmgrubgrugrupbgrupvgggrupvggbgrupvgggruvggbgruz$Type of encoder network architecturedefaulttypechoiceshelpz --elayerszNumber of encoder layersrrrz--eunitsz-ui,zNumber of encoder hidden unitsz--eprojs@z"Number of encoder projection unitsz --subsample1zhSubsample input frames x_y_z means subsample every x frame at 1st layer, every y frame at 2nd layer etc.) add_argumentstrintgroupr]/home/ubuntu/.local/lib/python3.10/site-packages/funasr/models/language_model/rnn/argument.py add_arguments_rnn_encoder_commons8r!c Cs|jddtddgdd|jddtdd |jd d td d |jd dtdd |jddtdd |jdddtdddgdd|S)z(Define common arguments for RNN decoder.z--dtyperr z$Type of decoder network architecturerz --dlayerszNumber of decoder layersrz--dunitsrzNumber of decoder hidden unitsz--dropout-rate-decoderzDropout rate for the decoderz--sampling-probabilityz-Ratio of predicted labels fed back to decoderz --lsm-type?unigramz8Apply label smoothing with a specified distribution type)constrrnargsrrrrrfloatrrrr add_arguments_rnn_decoder_common:s> r+cCs|jddtgddd|jddtdd |jd d td d |jd dtdd |jddtdd |jddtdd |jddtdd |S)z*Define common arguments for RNN attention.z--atypedot) noattr,addlocationcoveragecoverage_location location2dlocation_recurrentmulti_head_dotmulti_head_addmulti_head_locmulti_head_multi_res_loczType of attention architecturerz--adimrz-Number of attention transformation dimensionsrz--awinz$Window size for location2d attentionz--aheadsrz(Number of heads for multi head attentionz --aconv-chanszvNumber of attention convolution channels (negative value indicates no location-aware attention)z --aconv-filtsdzuNumber of attention convolution filters (negative value indicates no location-aware attention)z--dropout-rater#zDropout rate for the encoderr)rrrr "add_arguments_rnn_attention_common]sNr;N)__doc__r!r+r;rrrr s3 #