Print iterative loss for train and validation data on Scala

[Faisal]

Hi all, 

This might be a trivial, but I couldn't fins any documentation about it. I am trying to have the Optimizer.optimize to print the loss value for each iteration for both train and validation data.

I tried Optimizer.setTrainSummary(trainSummary), with trainSummary = TrainSummary(myLogDir, myAppName). This saves the results for visualization purposes, but it does not print them in my log. Is there a Scala equivalent to the Python model.get_train_summary? 

Thank all,

Faisal 

 

[Xin Qiu]

Hi, Faisal

You can find some related usage in our unit tests, see https://github.com/intel-analytics/BigDL/blob/411811681f24692d0b3631e328f0f4621529cc56/scala/dllib/src/test/scala/com/intel/analytics/bigdl/dllib/optim/DistriOptimizerSpec.scala#L523.

Bests,

-Xin