[TESTING] allclose fixup (#724)
This commit is contained in:
@@ -102,7 +102,6 @@ def allclose(x, y, tol=1e-2):
|
|||||||
diff = abs(x - y)
|
diff = abs(x - y)
|
||||||
x_max = torch.max(x)
|
x_max = torch.max(x)
|
||||||
y_max = torch.max(y)
|
y_max = torch.max(y)
|
||||||
tol = 1e-2
|
|
||||||
err = torch.max(diff) / torch.max(x_max, y_max)
|
err = torch.max(diff) / torch.max(x_max, y_max)
|
||||||
return err <= tol
|
return err <= tol
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user