Train-Validation Split Results in Zero Samples for Some Classes in PyTorch Dataset

Train-Validation Split Results in Zero Samples for Some Classes in PyTorch Dataset

Jul 29 '24 Comments: 1 Answers: 0

0

I'm working on an image classification using PyTorch. My dataset is in directory format. I've set up my data pipeline and model. Still, I'm encountering an issue with my train-validation split where some classes have zero samples in either the training or validation datasets. Here's the relevant part of my…

Open Full Question