本文整理汇总了Java中org.nd4j.linalg.learning.config.IUpdater.hasLearningRate方法的典型用法代码示例。如果您正苦于以下问题:Java IUpdater.hasLearningRate方法的具体用法?Java IUpdater.hasLearningRate怎么用?Java IUpdater.hasLearningRate使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类org.nd4j.linalg.learning.config.IUpdater
的用法示例。
在下文中一共展示了IUpdater.hasLearningRate方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Java代码示例。
示例1: setLearningRate
import org.nd4j.linalg.learning.config.IUpdater; //导入方法依赖的package包/类
private static void setLearningRate(MultiLayerNetwork net, int layerNumber, double newLr, ISchedule newLrSchedule, boolean refreshUpdater) {
Layer l = net.getLayer(layerNumber).conf().getLayer();
if (l instanceof BaseLayer) {
BaseLayer bl = (BaseLayer) l;
IUpdater u = bl.getIUpdater();
if (u != null && u.hasLearningRate()) {
if (newLrSchedule != null) {
u.setLrAndSchedule(Double.NaN, newLrSchedule);
} else {
u.setLrAndSchedule(newLr, null);
}
}
//Need to refresh the updater - if we change the LR (or schedule) we may rebuild the updater blocks, which are
// built by creating blocks of params with the same configuration
if (refreshUpdater) {
refreshUpdater(net);
}
}
}