Visual-Behavior / detr-tensorflow

Tensorflow implementation of DETR : Object Detection with Transformers
MIT License
168 stars 53 forks source link

Typo in transformer.py #11

Closed benqua closed 3 years ago

benqua commented 3 years ago

Hi,

There is a small typo in transformer.py, line 320:

        if attn_mask is not None:
            ann_output_weights += attn_mask

should be

        if attn_mask is not None:
            attn_output_weights += attn_mask
thibo73800 commented 3 years ago

Thanks !