.. automodule:: pytorch_ood.augment