guided-diffusion icon indicating copy to clipboard operation
guided-diffusion copied to clipboard

When training classifier_train.py with my own dataset, the loss never drops and the acc is very low!

Open Supzekun opened this issue 11 months ago • 1 comments

I trained the classifier on my own dataset, which has a total of 100 images divided into 5 classes with 20 images in each class. Each image is in the form of label1_xxx.jpg, label2_xxx.jpg named. But when training, the loss is always big and not decreasing, the acc is always low and my batch_size is 4. Why is this? Is the dataset too small?

Supzekun avatar Mar 08 '24 05:03 Supzekun