pulled SpatialClassNLLCriterion from torch