change loss.numpy()[0] to float(loss) to adapt 0D (#51)

PaddlePaddle · Jan 18, 2023 · f0cc490 · f0cc490
1 parent bef7bcf
commit f0cc490
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/competitions/mrc_utils.py b/competitions/mrc_utils.py
@@ -780,11 +780,11 @@ def hook(layer, input, output):
     probas_end.stop_gradient = True
 
     probas =[]
-    for idx in range(len(probas_start)):#(preds_start.numpy(), preds_end.numpy())
+    for idx in range(len(probas_start)):    #(preds_start.numpy(), preds_end.numpy())
         probas.append([probas_start[idx].numpy(), probas_end[idx].numpy()])
     preds =[]
-    for idx in range(len(preds_start)):#(preds_start.numpy(), preds_end.numpy())
-        preds.append([preds_start[idx].numpy()[0],preds_end[idx].numpy()[0]])
+    for idx in range(len(preds_start)):     #(preds_start.numpy(), preds_end.numpy())
+        preds.append([int(preds_start[idx]), int(preds_end[idx])])
     preds = np.array(preds)
     return gradients, preds, target_feature_map[0].numpy(), probas