I don't know if it is a problem with my environment, I just changed the dataset path to my own, but the error is reported. My torch==1.3.1.
error:
File "/home/ubuntu/.conda/envs/mmlab/lib/python3.7/site-packages/torch/nn/functional.py", line 3352, in multi_head_attention_forward attn_output_weights += attn_mask
RuntimeError: output with shape [16, 300, 300] doesn't match the broadcast shape [1, 16, 300, 300]