This commit is contained in:
Shreyansh Pathak 2023-05-22 17:39:07 -05:00 committed by GitHub
commit f6d44c1ad3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -70,7 +70,7 @@ def get_learning_rate_decay_fn(params):
global_step=global_step, global_step=global_step,
decay_steps=params.decay_steps, decay_steps=params.decay_steps,
end_learning_rate=params.end_learning_rate, end_learning_rate=params.end_learning_rate,
power=params.polynomial_power if 'polynomial_power' in paramsv else 1.0, power=params.polynomial_power if 'polynomial_power' in paramsv else 0.5,
) )
return polynomial_decay_fn return polynomial_decay_fn