Convert caffe AbsVal to mx.symbol.abs in caffe converter (#5984)

apache · Apr 26, 2017 · 56e712e · 56e712e
1 parent 02671eb
commit 56e712e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tools/caffe_converter/convert_symbol.py b/tools/caffe_converter/convert_symbol.py
@@ -176,6 +176,9 @@ def _parse_proto(prototxt_fname):
             need_flatten[name] = False
             param = layer[i].reshape_param
             param_string = "shape=(%s)" % (','.join(param.shape.dim),)
+        if layer[i].type == 'AbsVal': 
+        	type_string = 'mx.symbol.abs' 
+        	need_flatten[name] = need_flatten[mapping[layer[i].bottom[0]]] 
 
         if skip_layer:
             assert len(layer[i].bottom) == 1