From e823b310404829108d7ebf8f88fb75586e11f7d8 Mon Sep 17 00:00:00 2001 From: BBing <67486385@qq.com> Date: Wed, 4 Oct 2023 12:54:01 +0800 Subject: [PATCH] Update attn.py --- models/attn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/attn.py b/models/attn.py index 9569477..5e939ce 100644 --- a/models/attn.py +++ b/models/attn.py @@ -4,7 +4,7 @@ import mindspore.ops as ops from mindspore import Tensor from mindspore.common import dtype as mstype - +1 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__()