Parametrize the augmentation transformations to apply during training

Refs https://gitlab.com/teklia/atr/dan/-/merge_requests/171#note_1454987650

The list of applied transformations is currently hardcoded. We could instead try to parametrize it (with a default configuration and examples in the documentation).