Generalization to translation shifts: a study in architectures and augmentations