adding addtional weights parameter to ClassNLLCriterion, which is 1D tensor assigning different weight to each class