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