Bug: random_hflip function

[xiankgx]

With ref to: https://github.com/lucidrains/stylegan2-pytorch/blob/master/stylegan2_pytorch/stylegan2_pytorch.py#L389-L392

def random_hflip(tensor, prob):
    if prob > random():
        return tensor
    return torch.flip(tensor, dims=(3,))

Should prob be augment probability or non-augment probability? If we set prob=1.0, then we do NOT do horizontal flip 100% of the time.

[lucidrains]

@xiankgx oh shoot, yes you are right 🙏 0588213