Ghazi Felhi issues

Repositories
Issues
Comments

Results 1 issues of


                                            Ghazi Felhi

DPMultiheadAttention Doesn't check attention mask correctly when batch_first is set to True.

## 🐛 Bug This bug happens when using `DPMultiheadAttention` with the `batch_first` parameter set to True.[ While going through the shape checks](https://github.com/pytorch/opacus/blob/42b0e7275c394f08544bcc58040a7c2630b28c18/opacus/layers/dp_multihead_attention.py#L256), the forward method of `DPMultiheadAttention` check whether the...