We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0387b88 commit 872804fCopy full SHA for 872804f
1 file changed
test/common_utils.py
@@ -292,7 +292,8 @@ def compare(self) -> None:
292
actual, expected = self._equalize_attributes(actual, expected)
293
294
if self.mae:
295
- actual, expected = self._promote_for_comparison(actual, expected)
+ if actual.dtype is torch.uint8:
296
+ actual, expected = actual.to(torch.int), expected.to(torch.int)
297
mae = float(torch.abs(actual - expected).float().mean())
298
if mae > self.atol:
299
self._fail(
0 commit comments