Add File

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<float[][]> fistNervesModel;//FNN层第一层model
+    private List<float[][]> 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<float[][]> getFistNervesModel() {
+        return fistNervesModel;
+    }
+
+    public void setFistNervesModel(List<float[][]> fistNervesModel) {
+        this.fistNervesModel = fistNervesModel;
+    }
+
+    public List<float[][]> getSecondNervesModel() {
+        return secondNervesModel;
+    }
+
+    public void setSecondNervesModel(List<float[][]> secondNervesModel) {
+        this.secondNervesModel = secondNervesModel;
+    }
+
+    public LayNormModel getLineLayNormModel() {
+        return lineLayNormModel;
+    }
+
+    public void setLineLayNormModel(LayNormModel lineLayNormModel) {
+        this.lineLayNormModel = lineLayNormModel;
+    }
+}