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