search

zhutmost / lsq-net

Unofficial implementation of LSQ-Net, a neural network quantization framework
MIT License
276 stars 40 forks source link

Train and validation set are imbalanced #5

Open zhutmost opened 4 years ago

zhutmost commented 4 years ago

When using the random_split function provided in PyTorch to split the dataset to training and validation sets, the propotion of categories are imbalanced.

zhutmost commented 4 years ago

Fixed in commit https://github.com/zhutmost/lsq-net/commit/02f802e6652d3d067b953f756f01dd2d50d54456.

zhutmost commented 4 years ago

However, the imbalance among categories still exists. Maybe need another Sampler.