diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index d112793c711f7a21dcf6f8ee79a04a95badd1fd6..16ea7a7ddf725cd60df93c4442c00c308e58d254 100644 --- 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: