Skip to content

Commit 58cd6aa

Browse files
committed
fix median accuracy
1 parent 0730e74 commit 58cd6aa

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tester/base.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -482,6 +482,8 @@ def gen_paddle_input(self):
482482
self.paddle_args[3] = "gels"
483483
elif "driver" in self.paddle_kwargs:
484484
self.paddle_kwargs["driver"] = "gels"
485+
if self.api_config.api_name == "paddle.Tensor.median":
486+
self.paddle_kwargs["mode"] = "min"
485487

486488
if self.need_check_grad():
487489
if (self.api_config.api_name[-1] == "_" and self.api_config.api_name[-2:] != "__") or self.api_config.api_name == "paddle.Tensor.__setitem__":

0 commit comments

Comments
 (0)