Browse Source

fix: predict with multiple outputs

tags/v0.110.0-LSTM-Model
Kevin Hjelden 2 years ago
parent
commit
93cd2b66a6
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/TensorFlowNET.Keras/Engine/Model.Predict.cs

+ 2
- 2
src/TensorFlowNET.Keras/Engine/Model.Predict.cs View File

@@ -84,7 +84,7 @@ namespace Tensorflow.Keras.Engine
Steps = data_handler.Inferredsteps
});

Tensor batch_outputs = null;
Tensors batch_outputs = null;
_predict_counter.assign(0);
callbacks.on_predict_begin();
foreach (var (epoch, iterator) in data_handler.enumerate_epochs())
@@ -95,7 +95,7 @@ namespace Tensorflow.Keras.Engine
var tmp_batch_outputs = run_predict_step(iterator);
if (batch_outputs == null)
{
batch_outputs = tmp_batch_outputs[0];
batch_outputs = tmp_batch_outputs;
}
else
{


Loading…
Cancel
Save