未验证 提交 20075bf3 编写于 作者: R Roc 提交者: GitHub

Recover spectral norm self._eps (#50278)

上级 9a47b11d
...@@ -1818,7 +1818,7 @@ class SpectralNorm(Layer): ...@@ -1818,7 +1818,7 @@ class SpectralNorm(Layer):
): ):
super().__init__() super().__init__()
self._power_iters = power_iters self._power_iters = power_iters
self._epsilon = eps self._eps = eps
self._dim = dim self._dim = dim
self._dtype = dtype self._dtype = dtype
...@@ -1859,7 +1859,7 @@ class SpectralNorm(Layer): ...@@ -1859,7 +1859,7 @@ class SpectralNorm(Layer):
self.weight_v, self.weight_v,
self._dim, self._dim,
self._power_iters, self._power_iters,
self._epsilon, self._eps,
) )
check_variable_and_dtype( check_variable_and_dtype(
...@@ -1876,7 +1876,7 @@ class SpectralNorm(Layer): ...@@ -1876,7 +1876,7 @@ class SpectralNorm(Layer):
attrs={ attrs={
"dim": self._dim, "dim": self._dim,
"power_iters": self._power_iters, "power_iters": self._power_iters,
"eps": self._epsilon, "eps": self._eps,
}, },
) )
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册