Fixes TypeError: torch.Size() takes an iterable of 'int' (item 1 is '…#729

Rajathbharadwaj wants to merge 1 commit intoNVIDIA-Merlin:mainNVIDIA-Merlin/Transformers4Rec:mainfrom

Rajathbharadwaj:patch-1Rajathbharadwaj/Transformers4Rec:patch-1Copy head branch name to clipboard