diff --git a/src/main/java/org/dromara/easyai/transFormer/model/FirstDecoderModel.java b/src/main/java/org/dromara/easyai/transFormer/model/FirstDecoderModel.java new file mode 100644 index 0000000..456bceb --- /dev/null +++ b/src/main/java/org/dromara/easyai/transFormer/model/FirstDecoderModel.java @@ -0,0 +1,22 @@ +package org.dromara.easyai.transFormer.model; + +public class FirstDecoderModel { + private MultiSelfAttentionModel multiSelfAttentionModel;//注意力层model + private LayNormModel attentionLayNormModel;//残差1层model + + public MultiSelfAttentionModel getMultiSelfAttentionModel() { + return multiSelfAttentionModel; + } + + public void setMultiSelfAttentionModel(MultiSelfAttentionModel multiSelfAttentionModel) { + this.multiSelfAttentionModel = multiSelfAttentionModel; + } + + public LayNormModel getAttentionLayNormModel() { + return attentionLayNormModel; + } + + public void setAttentionLayNormModel(LayNormModel attentionLayNormModel) { + this.attentionLayNormModel = attentionLayNormModel; + } +}