| 137 | |
| 138 | |
| 139 | def reset_auc(use_fleet=False, auc_num=1): |
| 140 | # for static clear auc |
| 141 | auc_var_name = [] |
| 142 | for i in range(auc_num * 5): |
| 143 | auc_var_name.append("_generated_var_{}".format(i)) |
| 144 | |
| 145 | for name in auc_var_name: |
| 146 | param = paddle.static.global_scope().find_var(name) |
| 147 | if param == None: |
| 148 | continue |
| 149 | tensor = param.get_tensor() |
| 150 | if param: |
| 151 | tensor_array = np.zeros(tensor._get_dims()).astype("int64") |
| 152 | if use_fleet: |
| 153 | trainer_id = paddle.distributed.get_rank() |
| 154 | tensor.set(tensor_array, paddle.CUDAPlace(trainer_id)) |
| 155 | else: |
| 156 | tensor.set(tensor_array, paddle.CPUPlace()) |
| 157 | logger.info("AUC Reset To Zero: {}".format(name)) |
| 158 | |
| 159 | |
| 160 | def auc(stat_pos, stat_neg, scope, util): |