SwinUNETR: optional flash attention (scaled_dot_product_attention) in WindowAttention#8977
Open
aymuos15 wants to merge 2 commits into
Open
SwinUNETR: optional flash attention (scaled_dot_product_attention) in WindowAttention#8977aymuos15 wants to merge 2 commits into
aymuos15 wants to merge 2 commits into
