File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -4188,7 +4188,7 @@ local function firstElement(a)
41884188 return torch .type (a ) == ' table' and a [1 ] or a
41894189end
41904190
4191- function rnntest .MaskZero ()
4191+ function rnntest .MaskZero_main ()
41924192 local recurrents = {[' recurrent' ] = recurrentModule (), [' lstm' ] = lstmModule ()}
41934193 -- Note we use lstmModule input signature and firstElement to prevent duplicate code
41944194 for name , recurrent in pairs (recurrents ) do
@@ -4290,7 +4290,7 @@ function rnntest.MaskZero()
42904290 end
42914291end
42924292
4293- function rnntest .TrimZero ()
4293+ function rnntest .TrimZero_main ()
42944294 local recurrents = {[' recurrent' ] = recurrentModule (), [' lstm' ] = lstmModule ()}
42954295 -- Note we use lstmModule input signature and firstElement to prevent duplicate code
42964296 for name , recurrent in pairs (recurrents ) do
You can’t perform that action at this time.
0 commit comments