Commit a7b502f0 authored by berak's avatar berak

dnn: preserve name, type strings for ShuffleLayer

parent 0bb2c115
...@@ -14,6 +14,7 @@ public: ...@@ -14,6 +14,7 @@ public:
ShuffleChannelLayerImpl(const LayerParams& params) ShuffleChannelLayerImpl(const LayerParams& params)
{ {
group = params.get<int>("group", 1); group = params.get<int>("group", 1);
setParamsFrom(params);
} }
bool getMemoryShapes(const std::vector<MatShape> &inputs, bool getMemoryShapes(const std::vector<MatShape> &inputs,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment