diff --git a/src/main/java/org/dromara/easyai/transFormer/model/CodecBlockModel.java b/src/main/java/org/dromara/easyai/transFormer/model/CodecBlockModel.java new file mode 100644 index 0000000..86d7d66 --- /dev/null +++ b/src/main/java/org/dromara/easyai/transFormer/model/CodecBlockModel.java @@ -0,0 +1,52 @@ +package org.dromara.easyai.transFormer.model; + + +import java.util.List; + +public class CodecBlockModel { + private MultiSelfAttentionModel multiSelfAttentionModel;//注意力层model + private LayNormModel attentionLayNormModel;//残差1层model + private List fistNervesModel;//FNN层第一层model + private List secondNervesModel;//FNN层第二层model + private LayNormModel lineLayNormModel;//残差层最后2层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; + } + + public List getFistNervesModel() { + return fistNervesModel; + } + + public void setFistNervesModel(List fistNervesModel) { + this.fistNervesModel = fistNervesModel; + } + + public List getSecondNervesModel() { + return secondNervesModel; + } + + public void setSecondNervesModel(List secondNervesModel) { + this.secondNervesModel = secondNervesModel; + } + + public LayNormModel getLineLayNormModel() { + return lineLayNormModel; + } + + public void setLineLayNormModel(LayNormModel lineLayNormModel) { + this.lineLayNormModel = lineLayNormModel; + } +}