I am running into a similar issue. I’m running TS v2.7 and python 3.7.4
I’ve checked the shapes passing and the batch size as referred to above and those are not the issues. However I get it on every iteration.
Any help would be appreciated!