We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 9ae3a15 + 2fbb461 commit 543d026Copy full SHA for 543d026
tester/accuracy.py
@@ -383,7 +383,13 @@ def test(self):
383
if self.api_config.api_name == "paddle.scale":
384
paddle_out_grads = paddle_out_grads[0]
385
torch_out_grads = torch_out_grads[0]
386
-
+ if self.api_config.api_name == "paddle.nn.functional.binary_cross_entropy":
387
+ paddle_out_grads = paddle_out_grads[0]
388
+ torch_out_grads = torch_out_grads[0]
389
+ if self.api_config.api_name == "paddle.nn.functional.binary_cross_entropy_with_logits":
390
391
392
+
393
if isinstance(paddle_out_grads, paddle.Tensor):
394
if isinstance(torch_out_grads, torch.Tensor):
395
try:
0 commit comments