def METHOD_NAME(self, model_name):
    print(" - Loading pretrained BERT Mapping model")
    self.load_state_dict(
        torch.load(os.path.join(self.model_dir, self.dataset_dir, model_name), map_location=self.device))