Pytorch Print Loss Value