def METHOD_NAME(self, load_to_cpu=False):
    pretrained_dict = torch.load(
        self.model_path,
        map_location=torch.device('cuda')
        if torch.cuda.is_available() else torch.device('cpu'))
    self.net.load_state_dict(pretrained_dict['state_dict'], strict=False)
    self.net.eval()