Add Polynomial Decay based on epochs (intel-analytics#1575)

* poly on epoch * doc and ut * style
dding3 · Aug 23, 2019 · 7bfe7c1 · 7bfe7c1
1 parent a78a917
commit 7bfe7c1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/...c/test/scala/com/intel/analytics/bigdl/pipeline/api/keras/optimizers/OptimizersSpec.scala b/...c/test/scala/com/intel/analytics/bigdl/pipeline/api/keras/optimizers/OptimizersSpec.scala
@@ -66,6 +66,7 @@ class OptimizersSpec extends FlatSpec with Matchers with BeforeAndAfter {
       Default(),
       Plateau("Loss", 0.1f, 1, "min", 0.01f, 0, 1e-15f),
       Poly(0.1, 50),
+      PolyEpochDecay(3.0, 5),
       SequentialSchedule(5).add(Warmup(1e-4), 10).add(Default(), 40)
     ).foreach { schedule =>
       val classifier = NNClassifier(model, criterion, Array(6))