Browse Source

Resolve the comment.

pull/996/head
Yaohui Liu 2 years ago
parent
commit
bebf9b4fd2
No known key found for this signature in database GPG Key ID: E86D01E1809BD23E
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/TensorFlowNET.Keras/Engine/DataAdapters/TensorLikeDataAdapter.cs

+ 1
- 1
src/TensorFlowNET.Keras/Engine/DataAdapters/TensorLikeDataAdapter.cs View File

@@ -21,7 +21,7 @@ namespace Tensorflow.Keras.Engine.DataAdapters
{ {
this.args = args; this.args = args;
_process_tensorlike(); _process_tensorlike();
num_samples = (int)args.X[0].shape[0];
num_samples = (int)args.X.shape[0];
var batch_size = args.BatchSize == -1 ? 32 : args.BatchSize; var batch_size = args.BatchSize == -1 ? 32 : args.BatchSize;
_batch_size = batch_size; _batch_size = batch_size;
_size = Convert.ToInt32(Math.Ceiling(num_samples / (batch_size + 0.0f))); _size = Convert.ToInt32(Math.Ceiling(num_samples / (batch_size + 0.0f)));


Loading…
Cancel
Save