def _prepare(tensor):
    if (self.args.precision == 'half'):
        tensor = tensor.half()
    return tensor.to(device)
def _prepare(tensor):
    if (self.args.precision == 'half'):
        tensor = tensor.half()
    return tensor.to(device)