提交 fdde4eff 编写于 作者: Z zlx

modify some topo

上级 43771ad4
...@@ -32,7 +32,7 @@ namespace paddle { ...@@ -32,7 +32,7 @@ namespace paddle {
/** /**
* The static pruning hook * The static pruning hook
* Static means user specify a sparsity_ratio before training started, and the * Static means user specify a sparsity_ratio before training started, and the
* network will prune the parameters based on the sparsity_ratio. More deatils * network will prune the parameters based on the sparsity_ratio. More details
* can be found https://arxiv.org/pdf/1506.02626.pdf. * can be found https://arxiv.org/pdf/1506.02626.pdf.
*/ */
......
...@@ -84,7 +84,7 @@ class HookAttribute(object): ...@@ -84,7 +84,7 @@ class HookAttribute(object):
assert is_compatible_with( assert is_compatible_with(
self.sparsity_ratio, self.sparsity_ratio,
float), 'sparisity_ratio must be float type' float), 'sparisity_ratio must be float type'
assert self.sparsity_ratio <= 1 and self.sparsity_ratio >= 0, 'sparisity_ratio must be a float between [0, 1] ' assert self.sparsity_ratio <= 1 and self.sparsity_ratio >= 0, 'sparsity_ratio must be a float between [0, 1] '
def __call__(self): def __call__(self):
return ParameterHook(self.type, sparsity_ratio=self.sparsity_ratio) return ParameterHook(self.type, sparsity_ratio=self.sparsity_ratio)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册