what dou u mean by doing sigmoid, u only have two categories?
I see u said the number of labels is 1024, I don’t know if your categories are 1024, if so, u need to use softmax instead of sigmoid.
In other words, what is predict.size()