Hi @zeakey ,
Thank you very much for your code to give me a lot of help, but when I train your datasets of sk506, I got initial loss around 10000, is that normal? And second question is when I use the program of demo.py, the caffemodel is sk506_it14k.caffemodel, I got error is "cudasuccess(29 vs. 0) driver shutting down"? Have you ever met before? And I put the question on google, the answer told me to check the BVLC/caffe#2016. How should I do to solve the problem? Any advice would be helpful to me.
Thanks!!!
Samylee