Solved by phan_phan in post #4 
              
              
                Okay, it seems that you have a phantom dimension that you need to get rid of. 
The tensor inputs has shape [batch, 1, 224], you need to squeeze it to have [batch, 224] instead. 
Instead of calling outputs = model(inputs), try : 
outputs = model(inputs.squeeze(1))