def METHOD_NAME(self, is_best=False):
    # save personalized model locally
    model_weights = self.model.state_dict()
    save_dict = {"model": model_weights, "epoch": self.epoch_global}
    if is_best:
        save_dict.update({"best_metric": self.best_metric})
        torch.save(save_dict, self.best_model_file_path)
    else:
        torch.save(save_dict, self.model_file_path)