mirror of
https://github.com/Relintai/MLPP.git
synced 2024-11-12 10:15:01 +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"){
|
||||
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){
|
||||
|
Loading…
Reference in New Issue
Block a user