test=develop

6 years ago · 6259dba5bd
parent eab2e5e5d4
commit 6259dba5bd
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -7470,10 +7470,11 @@ def reorg(x, stride, name=None):
                x=data, stride=2)
    """

+    helper = LayerHelper("reorg", **locals())
+
    if not (isinstance(stride, int)):
        raise ValueError("stride must be a python Int")

-    helper = LayerHelper("reorg", **locals())
    if name is None:
        out = helper.create_tmp_variable(dtype=x.dtype)
    else: