Error while running train_step()

[nikita99]

I encounter following error while trying to train the model with experts main function.

---

InvalidArgumentError: ```
Traceback (most recent call last)
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]

This is resolved.