thank you time and time again!
You just singlehandedly saved my recognition system class 
I got another problem though, this is a different datasets than my previous question,
when I try to classify a class that has >6 classes, this datasets has 70, I get error
RuntimeError: CUDA error: device-side assert triggered

The dataset is small, only about 1.7gb, my VRAM is 6gb, I’m using gtx 1060, so I think it’s not a problem in my GPU right? What’s wrong?