TomatoGreen

suggestion - add line: eps=torch.tensor(1e-07, dtype=torch.float32) or eps=torch.tensor(1e-07, dtype=gt.dtype)