From f0cc490438d4619b97c1bcf26794ffcea4f0ce95 Mon Sep 17 00:00:00 2001 From: Zhou Wei <1183042833@qq.com> Date: Wed, 18 Jan 2023 18:55:50 +0800 Subject: [PATCH] change loss.numpy()[0] to float(loss) to adapt 0D (#51) --- competitions/mrc_utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/competitions/mrc_utils.py b/competitions/mrc_utils.py index 576a8e6..2840e5c 100644 --- 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