|
Java example source code file (NesterovsUpdater.java)
The NesterovsUpdater.java Java example source codepackage org.deeplearning4j.nn.updater; import org.deeplearning4j.nn.api.Layer; import org.deeplearning4j.nn.api.Updater; import org.deeplearning4j.nn.updater.aggregate.UpdaterAggregator; import org.nd4j.linalg.api.ndarray.INDArray; import org.nd4j.linalg.learning.GradientUpdater; import org.nd4j.linalg.learning.Nesterovs; /** * @author Adam Gibson */ public class NesterovsUpdater extends BaseUpdater { @Override public void init() { } @Override public GradientUpdater init(String variable, INDArray gradient, Layer layer) { Nesterovs nesterovs = (Nesterovs) updaterForVariable.get(variable); if(nesterovs == null) { nesterovs = new Nesterovs(layer.conf().getLayer().getMomentum(), layer.conf().getLearningRateByParam(variable)); updaterForVariable.put(variable,nesterovs); } return nesterovs; } @Override public UpdaterAggregator getAggregator(boolean addThis){ NesterovsAggregator ag = new NesterovsAggregator(); if(addThis) ag.aggregate(this); return ag; } protected static class NesterovsAggregator extends BaseUpdater.UpdaterAggregatorImpl { @Override public Updater getUpdater() { return setUpdaterState(new NesterovsUpdater()); } } } Other Java examples (source code examples)Here is a short list of links related to this Java NesterovsUpdater.java source code file: |
... this post is sponsored by my books ... | |
#1 New Release! |
FP Best Seller |
Copyright 1998-2021 Alvin Alexander, alvinalexander.com
All Rights Reserved.
A percentage of advertising revenue from
pages under the /java/jwarehouse
URI on this website is
paid back to open source projects.