Browse Source

Resolve the comment.

tags/v0.100.5-BERT-load
Yaohui Liu Haiping 2 years ago
parent
commit
e5837dc8b3
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;
_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;
_batch_size = batch_size;
_size = Convert.ToInt32(Math.Ceiling(num_samples / (batch_size + 0.0f)));


Loading…
Cancel
Save