Issue while training on Vocaset about faceformer

[brbernardo90]

Hey @ujjawalcse , take a look my colab, the trainning is running: https://colab.research.google.com/drive/1BjSd3RGkm8LSZDnxjOCVEB5f4g4PIURy?usp=sharing

I copied from yours and comment some things.

[ujjawalcse]

Thanks @brbernardo90
It need your access. Sent you an access request.
Please check it once.

[brbernardo90]

@ujjawalcse Ops, done! Thank you, your colab helped me a lot to start it.

[ujjawalcse]

yaa @brbernardo90 , It's running fine in Google Colab.
But It's not running fine on my local PC.
This is configuration of my pc,
Ubuntu 18.04,
torch 1.5.1+cu101 (Also tried the Torch 1.9 but got the same error)
transformers 4.6.1
GPU : 8GB RTX 2070 Super
RAM : 32 GB,

[ujjawalcse]

Got the training working properly now.
Actually, I was using previous FaceFormer repository cloned when it was newly introduced in local PC.
So, I was missing one parameter batch_first=True at line no. 81 in decoder_layer in faceformer.py

Thanks.