Some problems with building my own datasets on classificaion

[WZHAWZH]

Thank you very much for the code you provided.Recently, I have been studied the dataset you provided for classification. Mnist serves as the source domain data and usps serves as the target domain data.However, when I build my own data to replace mnist and usps data, there is always an error. I hope to get your help.Error is shown as follows:

   IndexError: index 149 is out of bounds for axis  0 with  size 1 

[lfjia]

Have you solved that problem?