transformer prediction not going through else part

[swift88-clone]

When we predict results for transformers, prediction is done through training part and there is not return value in test part