public NeuralNetConfiguration getInnerConf(NeuralNetConfiguration conf) { NeuralNetConfiguration nnc = conf.clone(); nnc.setLayer(layer); return nnc; }
@Override public Map<String, INDArray> init(NeuralNetConfiguration conf, INDArray paramsView, boolean initializeParams) { FrozenLayer fl = (FrozenLayer) conf.getLayer(); Layer innerLayer = fl.getLayer(); ParamInitializer initializer = innerLayer.initializer(); conf.setLayer(innerLayer); Map<String, INDArray> m = initializer.init(conf, paramsView, initializeParams); conf.setLayer(fl); return m; }
@Override public Map<String, INDArray> getGradientsFromFlattened(NeuralNetConfiguration conf, INDArray gradientView) { FrozenLayer fl = (FrozenLayer) conf.getLayer(); Layer innerLayer = fl.getLayer(); ParamInitializer initializer = innerLayer.initializer(); conf.setLayer(innerLayer); Map<String, INDArray> m = initializer.getGradientsFromFlattened(conf, gradientView); conf.setLayer(fl); return m; } }
Layer newLayerConf = new org.deeplearning4j.nn.conf.layers.misc.FrozenLayer(origLayerConf); newLayerConf.setLayerName(origLayerConf.getLayerName()); editedModel.getLayerWiseConfigurations().getConf(i).setLayer(newLayerConf);
currLayerVertex.getLayerConf().setLayer(newLayerConf);