intel-analytics/ipex-llm

How to use BigDL to achieve early stop function

gdg1212 opened this issue · 1 comments

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()