Great job! I would like to ask, what data was used for training the pretrained checkpoint(model_state_dict_swint_51.2ap.pt)?