Can we first train a model using default torch.Tensor, which is torch.FloatTensor,
and convert it to torch.HalfTensor for inference?
Or can we directly use torch.HalfTensor for training and inference?