-
Notifications
You must be signed in to change notification settings - Fork 3.2k
split op questions for model otimizer #1
Copy link
Copy link
Closed
Description
Hi, I ran across a problem when I use mo to convert a caffemodel, it shows 'LayerParameter' object has no attribute 'num_split' for split layer. However, as far as I am concerned, split layer in caffe is just a copy operation. So I think this sentence _out_shape[node.axis] = np.int64(input_node.shape[node.axis] / _node.pb.num_split) in mo/ops/split.py should be like this:
out_shape[node.axis] = np.int64(input_node.shape[node.axis])
You may mistake split layer as slice layer, I guess.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels