Browse Source

Update attn.py

main
BBing 2 years ago
parent
commit
c939c84c54
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      models/attn.py

+ 1
- 0
models/attn.py View File

@@ -4,6 +4,7 @@ import mindspore.ops as ops
from mindspore import Tensor
from mindspore.common import dtype as mstype


class FullAttention(nn.Module):
def __init__(self, mask_flag=True, factor=5, scale=None, attention_dropout=0.1, output_attention=False):
super(FullAttention, self).__init__()


Loading…
Cancel
Save