examples/research_projects/bert-loses-patience/pabee/modeling_pabee_bert.py/BertModelWithPabee/__init__
class BertModelWithPabee:    def __init__(self, config):
        super().__init__(config)

        self.encoder = BertEncoderWithPabee(config)

        self.init_weights()
        self.patience = 0
        self.inference_instances_num = 0
        self.inference_layers_num = 0

        self.regression_threshold = 0