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