o }oЩiL9у@s<ddlZddlZddlmZGddДdГZGddДdГZdS)щN)┌jasperc@seZdZeddДГZddДZejjddДГZ ejjddДГZ ejjd d ДГZejjddДГZejjd dДГZ ejjddДГZejjddДГZejjddДГZejjddДГZejjddДГZejjddДГZejjddДГZejjddДГZejjdd ДГZd!S)"┌TestJasperBlockcKs0tdddgddgdgdddddН }|а|б|S) Nщщщщ┌reluTF) ┌inplanes┌planes┌kernel_size┌repeat┌stride┌dilation┌ activation┌ conv_mask┌ separable┌se)┌dict┌update)┌kwargs┌baseйr·[/home/ubuntu/.local/lib/python3.10/site-packages/tests/collections/asr/test_jasper_block.py┌jasper_base_configsЎ z"TestJasperBlock.jasper_base_configcs*daЗfddД}|а|бtdksJВdS)Nrcst|ИГrtd7adSdS)Nr)┌ isinstance┌_MODULE_EXISTS)┌mй┌clsrr┌ _traverse+s ■z6TestJasperBlock.check_module_exists.._traverse)r┌apply)┌self┌modulerrrrr┌check_module_exists's z#TestJasperBlock.check_module_existscCєк|jddН}tjа|аdббГ}tjd i|дd|iдО}tаdddб}tаdgб}||g|fГ\}}t |tjГs:JВ|dj tаd|ddgбksKJВ|ddksSJВdS) NFй┌residualrrrщГrr rйrr┌jasper_activations┌get┌pop┌JasperBlock┌torch┌randn┌tensorr┌shape┌Sizeйr!┌config┌act┌block┌x┌xlen┌y┌ylenrrr┌test_basic_block3є"z TestJasperBlock.test_basic_blockcCr$) NTr%rrrr'rr rr(r2rrr┌test_residual_blockBr;z#TestJasperBlock.test_residual_blockcCs╛|jdddН}tjа|аdббГ}tjdi|дd|iдО}tаdddб}tаdgб}||g|fГ\}}t |tjГs;JВ|dj tаd|d dgбksLJВ|ddksTJВt|j Гd ks]JВdS)NFщ)r&rrrrr'rr щ rйrrr)r*r+r,r-r.r/rr0r1┌len┌mconvr2rrr┌test_basic_block_repeatQs"z'TestJasperBlock.test_basic_block_repeatcCs┬|jdddgdН}tjа|аdббГ}tjd i|дd|iдО}tаdddб}tаdgб}||g|fГ\}}t |tjГs=JВ|d j tаd|d dgбksNJВ|d dksVJВt|j Гdks_JВdS)NFr=щ)r&rr rrrr'rr щr>rr?r2rrr┌test_basic_block_repeat_strideas"z.TestJasperBlock.test_basic_block_repeat_stridecCs─|jdddgddН}tjа|аdббГ}tjdi|дd|iдО}tаddd б}tаd gб}||g|fГ\}}t |tjГs>JВ|d j tаd|ddgбksOJВ|d dksWJВt|j Гd ks`JВdS)NFr=rCT)r&rr ┌stride_lastrrrr'rr щBr>rr?r2rrr┌#test_basic_block_repeat_stride_lastqs"z3TestJasperBlock.test_basic_block_repeat_stride_lastcCs└|jddddН}tjа|аdббГ}tjdi|дd|iдО}tаdddб}tаdgб}||g|fГ\}}t |tjГstest_blocks_with_different_input_output_channels_conv_residualbrШzPTestParallelBlock.test_blocks_with_different_input_output_channels_conv_residualc CsАd}d}|j||dНg}tа|б}tаd|dб}tаdgб}||g|fГ\}}|djtаd|dgбks6JВ|ddks>JВdS)NrrrГrrЖr'r)rВrrКr-r.r/r0r1) r!rНrОrМr5r6r7r8r9rrr┌test_single_blockrs z#TestParallelBlock.test_single_blockc CsАg}d}d}tdГD]}|а|j||dНбq tj|dddН}tаd|dб}tаd gб}||g|fГ\}}tjа |d |бdS)NrrCrГ┌dropoutgЁ?)┌aggregation_mode┌block_dropout_probrrЖr'r) rИrЙrВrrКr-r.r/┌testing┌assert_close) r!rМrНrОrПr5r6r7r8rrr┌test_tower_dropoutАsz$TestParallelBlock.test_tower_dropoutN)ryrzr{r|rВr}r~rrРrУrЧrЩrЪrаrrrrrА+s rА)r}r-┌%nemo.collections.asr.parts.submodulesrrrАrrrr┌s