Browse Source

Merge branch 'tf.keras-0.3.image-classification'

tags/keras_v0.3.0
Oceania2018 4 years ago
parent
commit
8e4a98a282
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      src/TensorFlowNET.Core/Data/DatasetV2.cs

+ 3
- 1
src/TensorFlowNET.Core/Data/DatasetV2.cs View File

@@ -132,7 +132,9 @@ namespace Tensorflow
break;
}

yield return (results[0], results.Length == 1 ? null : results[1]);
yield return results.Length == 2
? (results[0], results[1])
: (null, results[0]);
}
}



Loading…
Cancel
Save