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