Make augmentation transforms part of the model

src/ptbench/configs/models/pasa.py

src/ptbench/data/datamodule.py

src/ptbench/data/shenzhen/default.py

src/ptbench/models/pasa.py

src/ptbench/scripts/train.py