记录一次联邦学习FedAvg算法实现的debug过程

过程

在tensorflow federated框架下的FedAvg实验中，一段代码总是捕捉到异常退出：

for predicted_y in batch_predicted_y:
   max = -1000
   flag = -1
   for j in range(10):
       if predicted_y[j] > max:
           max = predicted_y[j]
           flag = j
   if(flag==-1):
       sys.exit()