Closed coh-racng closed 5 years ago
Excellent find! Great error. Yes your workaround is fine for personal usage. However I don't really want to change the batchsize in the code just because of that.
I guess we should just ignore batches of size < 3 in the training. If we have more than 1 epoch, the RandomSampler will likely sample the heldout cells in a big batch after, so not a big deal. What do you think @romain-lopez , @gabmis ?
that makes sense to me !
Issue similar to #221 My dataset has 57345 cells and it so happens that 57345%128 == 1. so I end up with a batch that has size 1 during training, raising an error
Potential fix by adjusting batch size