How to use BigDL to achieve early stop function
gdg1212 opened this issue · 1 comments
gdg1212 commented
val model = Sequential[Float]()
.add(Dense(50, inputShape = Shape(inputSize)))
.add(Activation("relu"))
.add(Dense(25))
.add(Activation("relu"))
.add(Dense(1))
val optimizer = Optimizer(model = model,
sampleRDD = trainingTensor.map(_._2),
criterion = MSECriterion[Float](),
batchSize = 150000)
optimizer
.setOptimMethod(new Adam(0.001))
.setEndWhen(Trigger.maxEpoch(100))
.optimize()
qiuxin2012 commented
You can use something like .setEndWhen(Or(Trigger.maxEpoch(100), Trigger.MinLoss(0.0001)))
.
https://github.com/intel-analytics/BigDL-2.x/blob/b4cde76e58c3b97c69424f0c8bbc4c6d08879195/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/optim/ZooTrigger.scala