examples/research_projects/bert-loses-patience/pabee/modeling_pabee_bert.py/BertModelWithPabee/reset_stats
class BertModelWithPabee:    def reset_stats(self):
        self.inference_instances_num = 0
        self.inference_layers_num = 0