本文整理匯總了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);
}
}
}