hsoule

For anyone else having the same problem, the `nn.View` module in the model file you're loading `(alexnetowtbn.lua, alexnet.lua, ...)` is the problem. The input to it is of dimensions `batch_size...

Thanks for the clarification! It seems like dataset.lua loads the directories in a random order. I'm thinking of checking the order in which the different classes are being loaded and...

Absolutely. I was looking at the forceClasses variable in dataset.lua which could help with that, but I've already spent significant time training the models up to the 37th epoch, so...