@Override protected void postInit() { super.postInit(); if (accumulator == null) { log.warn("GradientsAccumulator is undefined, gradients sharing will be skipped"); return; } // just pass accumulator down the hill if (replicatedModel instanceof ComputationGraph) { ((ComputationGraph) replicatedModel).setGradientsAccumulator(accumulator); } else if (replicatedModel instanceof MultiLayerNetwork) { ((MultiLayerNetwork) replicatedModel).setGradientsAccumulator(accumulator); } // need to attach this device id to accumulator's workspaces accumulator.touch(); }
@Override protected void postInit() { super.postInit(); if (accumulator == null) { log.warn("GradientsAccumulator is undefined, gradients sharing will be skipped"); return; } // just pass accumulator down the hill if (replicatedModel instanceof ComputationGraph) { ((ComputationGraph) replicatedModel).setGradientsAccumulator(accumulator); } else if (replicatedModel instanceof MultiLayerNetwork) { ((MultiLayerNetwork) replicatedModel).setGradientsAccumulator(accumulator); } // need to attach this device id to accumulator's workspaces accumulator.touch(); }