checkpoint

This commit is contained in:
Joseph Redmon
2016-06-20 13:18:59 -07:00
parent ab75d5c578
commit e7072b8489
7 changed files with 42 additions and 4 deletions

View File

@ -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);