demo.py

Question

Sylvia-Yuju opened this issue a year ago · comments

When running the demo.py code, the following problem occurs, How can I solve

Aiedails · Answer 1 · Fri Mar 31 2023 10:08:25 GMT+0800 (China Standard Time)

Me too. So I'm simply let tgt_mask = tgt_mask.unsqueeze(0)and it works.

Sylviarrran · Answer 2 · Fri Mar 31 2023 10:12:46 GMT+0800 (China Standard Time)

Me too. So I'm simply let tgt_mask = tgt_mask.unsqueeze(0)and it works.

Yes, I solved it like this too