output of 17th cell of my linked notebook here : Google Colab giving this error after all epoch training is finish.
if you check the visualized image attached in the last code cell of that notebook then you will see that my model is able to detect all those 3 classes perfectly and with good performance but it’s saying all those 3 classes are “name”
i printed names in train.py file right before calling it and this dictionary was printed [“name”,“age”,“mobile”]