* torch.cuda.synchronize() update * torch.cuda.synchronize() update * torch.cuda.synchronize() update * newline