Solved by guen_gn in post #8 
              
              
                @ptrblck, that was awesome. After adding some list test and splitting data further into train,valid and test, it got approx. 88% accuracy. 
Running on list of single batch got all correct: 

# make a single prediction
print("Making prediction...")

enum_test_dl = list(enumerate(test_dl))
enum_test_d…