Don't overwrite loss tensor with loss value

This commit is contained in:
Reuben Morais 2018-11-13 19:16:35 -02:00
Родитель 81e62d0259
Коммит 0c6e695138
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -149,7 +149,7 @@ def evaluate(test_data, inference_graph, alphabet):
labels = pad_to_dense(batch['transcript'].values)
label_lengths = batch['transcript_len'].values
logits, loss = session.run([transposed, loss], feed_dict={
logits, loss_ = session.run([transposed, loss], feed_dict={
inputs['input']: features,
inputs['input_lengths']: features_len,
labels_ph: labels,
@ -157,7 +157,7 @@ def evaluate(test_data, inference_graph, alphabet):
})
logitses.append(logits)
losses.extend(loss)
losses.extend(loss_)
ground_truths = []
predictions = []