Update ClassNLLCriterion.lua