I am using the mean of features for training the network. It generates the same error, sometimes after 4 epochs, sometimes after 10 epochs or 19 epochs. I ve faced this type of probles first time. Any help in this regard will be appreciated.
Error after 19 epochs of training:
File “main.py”, line 510, in train
Here 6 is the index of class.