From b613fc8f910c4092307e5318058cfa75d9a1c8a0 Mon Sep 17 00:00:00 2001 From: BBing <67486385@qq.com> Date: Wed, 4 Oct 2023 12:53:38 +0800 Subject: [PATCH] Update decoder.py --- models/decoder.py | 1 + 1 file changed, 1 insertion(+) diff --git a/models/decoder.py b/models/decoder.py index 85970b1..dc93fe4 100644 --- a/models/decoder.py +++ b/models/decoder.py @@ -1,6 +1,7 @@ import mindspore.nn as nn import mindspore.ops.operations as ops + class DecoderLayer(nn.Cell): def __init__(self, self_attention, cross_attention, d_model, d_ff=None, dropout=0.1, activation="relu"):