examples/research_projects/bert-loses-patience/pabee/modeling_pabee_bert.py/BertModelWithPabee/set_patience
class BertModelWithPabee:    def set_patience(self, patience):
        self.patience = patience