in ()
7 # Training set
8 for images_batch, targets_batch in train_dataset.batch(batch_size):
----> 9 train_step(images_batch, targets_batch)
10 template = '\r Batch {}/{}, Loss: {}, Accuracy: {}'
11 print(template.format()
----------------------------------------------------------------------------
**InvalidArgumentError:** Received a label value of 26 which is outside the valid range of [0, 4). Label values: 21 21 2 2 17 17 15 15 18 18 26 26 6 6 9 9 6 6 12 12 8 8 16 16 3 3 20 20 6 6 17 17
[[node sparse_categorical_crossentropy/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits (defined at :7) ]] [Op:__inference_train_step_2751]
Errors may have originated from an input operation.
Input Source operations connected to node sparse_categorical_crossentropy/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits:
conv3d_model/output/BiasAdd (defined at :31)
Function call stack:
train_step
---
nikita99
commented
4 years ago
I encounter following error while trying to train the model with experts main function.
InvalidArgumentError: ``` Traceback (most recent call last)