Yes, it does. The algorithm works as expected. It’s just weird I get better results with batch size 1. Even in the paper you mentioned they suggest between 32-512, so that’s why I am dazzled. So, I was thinking maybe I have an error on my code or there is something I am ignoring.