Fixed sizeAverage for NCECriterion

Element-Research / rnn

Recurrent Neural Network library for Torch7's nn

BSD 3-Clause "New" or "Revised" License

939 stars 313 forks source link

Closed danpol closed 7 years ago

danpol commented 7 years ago

NCECriterion returns table as gradInput, so you can't just use :div(nStep) for size averaging

nicholas-leonard commented 7 years ago

All tests pass. Thanks @danpol