Browse Source

Merge pull request #1065 from fryguy1013/fix-multiple-outputs

fix: predict with multiple outputs
tags/v0.110.0-LSTM-Model
Rinne GitHub 2 years ago
parent
commit
cd64ea96f3
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
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