public void setLayerParamLR(String variable) { double lr = layer.getLearningRateByParam(variable); double l1 = layer.getL1ByParam(variable); if (Double.isNaN(l1)) l1 = 0.0; //Not set double l2 = layer.getL2ByParam(variable); if (Double.isNaN(l2)) l2 = 0.0; //Not set learningRateByParam.put(variable, lr); l1ByParam.put(variable, l1); l2ByParam.put(variable, l2); }