fixing small cuda typing issue for ClassNLLCriterion