You need a type that is not affected by Data augmentation since transforms are always applied and dispatched. We’ll probably add this pretty soon, in the meantime, you can write a custom ItemTransform that takes a batch b and only applies RandomErasing on x.
3 Likes