mirror of
https://github.com/Relintai/MLPP.git
synced 2025-02-04 15:55:53 +01:00
learning rate scheduler- warning with returns fixed
This commit is contained in:
parent
38c216c68f
commit
43ca8aa5a7
@ -566,6 +566,7 @@ void ANN::Adam(double learning_rate, int max_epoch, int mini_batch_size, double
|
|||||||
else if(lrScheduler == "Epoch"){
|
else if(lrScheduler == "Epoch"){
|
||||||
return learningRate * (decayConstant / std::sqrt(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){
|
void ANN::addLayer(int n_hidden, std::string activation, std::string weightInit, std::string reg, double lambda, double alpha){
|
||||||
|
Loading…
Reference in New Issue
Block a user