Port automatic augmentation probability from lightweight gan.

[tannisroot]

A very handy feature, would be lovely to have it in this version as well :D

[lucidrains]

@tannisroot ok done! https://github.com/lucidrains/stylegan2-pytorch/releases/tag/1.5.10

[tannisroot]

Thanks!

[tannisroot]

Nitpicking here, but I've noticed that it doesn't report that it enabled augmentation like lightweight gan does. A very small feature but give you piece of mind that the feature actually works.

[lucidrains]

@tannisroot oh that's weird, it should be at this line https://github.com/lucidrains/stylegan2-pytorch/blob/master/stylegan2_pytorch/stylegan2_pytorch.py#L869