Error while running model.py

[seoho-kang]

RuntimeError: The size of tensor a (96) must match the size of tensor b (128) at non-singleton dimension 0

Error above, keep occurs. My environment is python 2.7, torch 0.3.0 post4.

[rythei]

It was an error with the batch sizes changing at the end of each epoch, should work now (though I'm using Python 3.5).