From 2f7f29e1529d924c1cd7a7b380e443e79a6f8e5e Mon Sep 17 00:00:00 2001 From: b_rookie Date: Thu, 14 Aug 2025 16:31:17 +0800 Subject: [PATCH 1/2] fix --- mindscience/models/transformer/__init__.py | 4 ++-- tests/models/transformer/test_attention.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mindscience/models/transformer/__init__.py b/mindscience/models/transformer/__init__.py index eab6d08d5..b7300a82d 100644 --- a/mindscience/models/transformer/__init__.py +++ b/mindscience/models/transformer/__init__.py @@ -13,7 +13,7 @@ # limitations under the License. # ============================================================================ """init""" -from .attention import MultiHeadAttention, TransformerBlock +from .attention import Attention, MultiHeadAttention, TransformerBlock from .vit import ViT -__all__ = ["MultiHeadAttention", "TransformerBlock", "ViT"] +__all__ = ["Attention", "MultiHeadAttention", "TransformerBlock", "ViT"] diff --git a/tests/models/transformer/test_attention.py b/tests/models/transformer/test_attention.py index 60ed1d3b4..637bbfd67 100644 --- a/tests/models/transformer/test_attention.py +++ b/tests/models/transformer/test_attention.py @@ -22,7 +22,7 @@ from mindspore import Tensor, ops, load_checkpoint, load_param_into_net, jit_cla from mindspore import dtype as mstype from mindscience.models import Attention, MultiHeadAttention, TransformerBlock, DropPath, ViT -from mindscience.models import RelativeRMSELoss +from mindscience.common import RelativeRMSELoss PROJECT_ROOT = os.path.abspath(os.path.join( os.path.dirname(__file__), "../../")) -- Gitee From d9b8e52e28888cc992cbc715c999f5096675d73e Mon Sep 17 00:00:00 2001 From: b_rookie Date: Thu, 14 Aug 2025 17:27:03 +0800 Subject: [PATCH 2/2] fix --- NOTICE | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/NOTICE b/NOTICE index 5bde55ddd..ee055cf6f 100644 --- a/NOTICE +++ b/NOTICE @@ -1,2 +1,2 @@ MindSpore MindScience -Copyright 2019-2021 Huawei Technologies Co., Ltd \ No newline at end of file +Copyright 2019-2025 Huawei Technologies Co., Ltd \ No newline at end of file -- Gitee