RuntimeError: The size of the 3D attn_mask is not correct.

Doubiiu / CodeTalker

[CVPR 2023] CodeTalker: Speech-Driven 3D Facial Animation with Discrete Motion Prior

MIT License

515 stars 57 forks source link

Closed RujingD closed 1 year ago

RujingD commented 1 year ago

你好！在训练stage2的时候对于nn.TransformerDecoder报错：RuntimeError: The size of the 3D attn_mask is not correct. 用来训练的数据最后一个维度不是15069，而是171，需要修改什么参数吗？

RujingD commented 1 year ago

I solved it