8000 refactor model.evaluate to deal with confilict · SciSharp/TensorFlow.NET@35d2e10 · GitHub
[go: up one dir, main page]

Skip to content

Commit 35d2e10

Browse files
refactor model.evaluate to deal with confilict
1 parent cb31cf4 commit 35d2e10

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/TensorFlowNET.Keras/Engine/Model.Evaluate.cs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ public Dictionary<string, float> evaluate(IEnumerable<Tensor> x, Tensor y, int v
7272
{
7373
var data_handler = new DataHandler(new DataHandlerArgs
7474
{
75-
X = new Tensors(x),
75+
X = new Tensors(x.ToArray()),
7676
Y = y,
7777
Model = this,
7878
StepsPerExecution = _steps_per_execution
@@ -168,7 +168,8 @@ Dictionary<string, float> test_function(DataHandler data_handler, Tensor[] data)
168168
Dictionary<string, float> test_step_multi_inputs_function(DataHandler data_handler, Tensor[] data)
169169
{
170170
var x_size = data_handler.DataAdapter.GetDataset().FirstInputTensorCount;
171-
var outputs = train_step(data_handler, new Tensors(data.Take(x_size)), new Tensors(data.Skip(x_size)));
171+
var outputs = train_step(data_handler, new Tensors(data.Take(x_size).ToArray()), new Tensors(data.Skip(x_size).ToArray()));
172+
tf_with(ops.control_dependencies(new object[0]), ctl => _train_counter.assign_add(1));
172173
return outputs;
173174
}
174175
}

0 commit comments

Comments
 (0)
0