learning rate scheduler- warning with returns fixed

This commit is contained in:
novak_99 2022-01-30 13:13:50 -08:00
parent 38c216c68f
commit 43ca8aa5a7
2 changed files with 1 additions and 0 deletions

View File

@ -566,6 +566,7 @@ void ANN::Adam(double learning_rate, int max_epoch, int mini_batch_size, double
else if(lrScheduler == "Epoch"){
return learningRate * (decayConstant / std::sqrt(epoch));
}
return learningRate;
}
void ANN::addLayer(int n_hidden, std::string activation, std::string weightInit, std::string reg, double lambda, double alpha){

BIN
a.out

Binary file not shown.