mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
checkpoint
This commit is contained in:
@ -64,6 +64,7 @@ float get_current_rate(network net)
|
||||
case EXP:
|
||||
return net.learning_rate * pow(net.gamma, batch_num);
|
||||
case POLY:
|
||||
if (batch_num < net.burn_in) return net.learning_rate * pow((float)batch_num / net.burn_in, net.power);
|
||||
return net.learning_rate * pow(1 - (float)batch_num / net.max_batches, net.power);
|
||||
case RANDOM:
|
||||
return net.learning_rate * pow(rand_uniform(0,1), net.power);
|
||||
|
Reference in New Issue
Block a user