Dataset are images of pills, where some of them are defect (heve some cracks) and some of them are correct. The task I want to solve is to find whether this pill is broken.
I tried to play with hp when use SGD, and it didn’t help. I will try ADAM now, thanks!
(I hoped that this network works, because it quite similar to MNIST task).
I am just not sure, that I did Dataset object right, because I did it first time. Can it be the problem?