Set frame_mask and cls_attn_mask if mask is None

lucidrains / TimeSformer-pytorch

Implementation of TimeSformer from Facebook AI, a pure attention-based solution for video classification

MIT License

686 stars 86 forks source link

Closed quaertym closed 3 years ago

quaertym commented 3 years ago

frame_mask and cls_attn_mask are used before definition if mask is None. This fixes that.

lucidrains commented 3 years ago

Oops 🙏 ty ty