diff --git a/src/main/java/org/dromara/easyai/transFormer/model/TransFormerModel.java b/src/main/java/org/dromara/easyai/transFormer/model/TransFormerModel.java new file mode 100644 index 0000000..a3ad88f --- /dev/null +++ b/src/main/java/org/dromara/easyai/transFormer/model/TransFormerModel.java @@ -0,0 +1,51 @@ +package org.dromara.easyai.transFormer.model; + +import java.util.List; + +public class TransFormerModel { + private List encoderBlockModels;//编码器模块 + private List decoderBlockModels;//解码器模块 + private FirstDecoderModel firstDecoderBlockModel;//第一个解码器模块 + private LineBlockModel lineBlockModel;//线性分类层 + private TransWordVectorModel transWordVectorModel;//词向量模型 + + public TransWordVectorModel getTransWordVectorModel() { + return transWordVectorModel; + } + + public void setTransWordVectorModel(TransWordVectorModel transWordVectorModel) { + this.transWordVectorModel = transWordVectorModel; + } + + public List getEncoderBlockModels() { + return encoderBlockModels; + } + + public void setEncoderBlockModels(List encoderBlockModels) { + this.encoderBlockModels = encoderBlockModels; + } + + public List getDecoderBlockModels() { + return decoderBlockModels; + } + + public void setDecoderBlockModels(List decoderBlockModels) { + this.decoderBlockModels = decoderBlockModels; + } + + public FirstDecoderModel getFirstDecoderBlockModel() { + return firstDecoderBlockModel; + } + + public void setFirstDecoderBlockModel(FirstDecoderModel firstDecoderBlockModel) { + this.firstDecoderBlockModel = firstDecoderBlockModel; + } + + public LineBlockModel getLineBlockModel() { + return lineBlockModel; + } + + public void setLineBlockModel(LineBlockModel lineBlockModel) { + this.lineBlockModel = lineBlockModel; + } +}