|
Java example source code file (AdaGradUpdater.java)
The AdaGradUpdater.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.AdaGrad; import org.nd4j.linalg.learning.GradientUpdater; /** * Ada grad updater * * @author Adam Gibson */ public class AdaGradUpdater extends BaseUpdater { @Override public void init() { } @Override public GradientUpdater init(String variable, INDArray gradient, Layer layer) { AdaGrad adaGrad = (AdaGrad) updaterForVariable.get(variable); if(adaGrad == null) { adaGrad = new AdaGrad(layer.conf().getLearningRateByParam(variable)); updaterForVariable.put(variable, adaGrad); } return adaGrad; } @Override public UpdaterAggregator getAggregator(boolean addThis){ AdaGradAggregator ag = new AdaGradAggregator(); if(addThis) ag.aggregate(this); return ag; } protected static class AdaGradAggregator extends BaseUpdater.UpdaterAggregatorImpl { @Override public Updater getUpdater() { return setUpdaterState(new AdaGradUpdater()); } } } Other Java examples (source code examples)Here is a short list of links related to this Java AdaGradUpdater.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.