Pytorch Class Weight Percentage