Hi, thanks very much for your reply. Your way works.
Because my labels original are string like: [00, 01, 11, 15, …, 45], is there any way that I can use my own image loader to convert those string labels to int to use them in crossentropy function withour directly assign the string to 0-16? Thanks?