Skip to content

Commit ec8ef93

Browse files
authored
fix eval for UIE (#3753)
1 parent 3ca3547 commit ec8ef93

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model_zoo/uie/evaluate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def evaluate(model, metric, data_loader, multilingual=False):
4545
else:
4646
start_prob, end_prob = model(batch["input_ids"],
4747
batch["token_type_ids"],
48-
batch["att_mask"], batch["pos_ids"])
48+
batch["pos_ids"], batch["att_mask"])
4949

5050
start_ids = paddle.cast(batch["start_positions"], 'float32')
5151
end_ids = paddle.cast(batch["end_positions"], 'float32')

0 commit comments

Comments
 (0)