We provide the pretrained SSAT weights used in our best experiments. Download here: weights/SSAT_asia.pth Please refer to the training configuration in parameter_exploration.py and evaluation scripts ...