From 69bea128fcd7d06a130160d66a6f2c33e4bfd5e9 Mon Sep 17 00:00:00 2001 From: PENGUINLIONG Date: Fri, 20 Oct 2017 01:49:40 +0800 Subject: [PATCH] Fixed attribute parameter (#184) --- python/nnvm/frontend/coreml.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/nnvm/frontend/coreml.py b/python/nnvm/frontend/coreml.py index e89c784dfe70..06bf80648da6 100644 --- a/python/nnvm/frontend/coreml.py +++ b/python/nnvm/frontend/coreml.py @@ -246,7 +246,7 @@ def ConcatLayerParams(op, insyms, symtab): insyms = [insyms] if op.sequenceConcat: raise NotImplementedError("Sequence Concat not supported") - ret = _sym.concatenate(*insyms, dim=1) + ret = _sym.concatenate(*insyms, axis=1) return ret def FlattenLayerParams(op, insym, symtab):