Model returns logits as tuple

[kirilltobola]

Model could return logits as tuple(tensor), except for just tensor.