Hello, I test with your code for 10 categories images classification, but I get an error, do you know how to fix it?
I think in order to train a model for more than 2 categories, I need to modifiy some variables or parameters. Could you give me some hints? Thanks a lot.
Traceback (most recent call last):
File "<ipython-input-6-96478bbedd1c>", line 271, in <module>
main()
File "<ipython-input-6-96478bbedd1c>", line 268, in main
train_network(g, batch_size, num_epochs, pb_file_path)
File "<ipython-input-6-96478bbedd1c>", line 254, in train_network
total_cost_in_train_set))
IndexError: tuple index out of range
Hello, I test with your code for 10 categories images classification, but I get an error, do you know how to fix it?
I think in order to train a model for more than 2 categories, I need to modifiy some variables or parameters. Could you give me some hints? Thanks a lot.