Hey @nuggimane,
I ended up implementing the crops as transforms and modifying the batch transforms of the validation dataloader. You can find it here.
Hope it helps.