RuntimeError: stack expects each tensor to be equal size but got

vision
1

EDIT: topic fixed, I found the issue and it was a bug somewhere else causing this to break.