I am using movielens datasets , where movie id, user id and genre are my input features. I put them in sequence and then created schema, not did any item regency and all.
Schema created but getting error while training model using Xlnet

I am getting the following error

torch.Size() takes an iterable of 'int' (item 1 is 'NoneType')

Please help me to resolve it
I am using transformer4rec in my thesis work and deadline is near,