diff --git a/src/TensorFlowNET.Core/Keras/Engine/ICallback.cs b/src/TensorFlowNET.Core/Keras/Engine/ICallback.cs index 07b93a75..296c32ac 100644 --- a/src/TensorFlowNET.Core/Keras/Engine/ICallback.cs +++ b/src/TensorFlowNET.Core/Keras/Engine/ICallback.cs @@ -13,5 +13,6 @@ public interface ICallback void on_predict_batch_end(long end_step, Dictionary logs); void on_predict_end(); void on_test_begin(); + void on_test_batch_begin(long step); void on_test_batch_end(long end_step, IEnumerable<(string, Tensor)> logs); } diff --git a/src/TensorFlowNET.Keras/Engine/Model.Evaluate.cs b/src/TensorFlowNET.Keras/Engine/Model.Evaluate.cs index 8e15428e..5397de47 100644 --- a/src/TensorFlowNET.Keras/Engine/Model.Evaluate.cs +++ b/src/TensorFlowNET.Keras/Engine/Model.Evaluate.cs @@ -72,12 +72,13 @@ namespace Tensorflow.Keras.Engine foreach (var step in data_handler.steps()) { - callbacks.on_train_batch_begin(step); + callbacks.on_test_batch_begin(step); logs = test_function(data_handler, iterator); var end_step = step + data_handler.StepIncrement; callbacks.on_test_batch_end(end_step, logs); } } + Console.WriteLine(); GC.Collect(); GC.WaitForPendingFinalizers(); }