From 6d46498b5d34e4ae53cd62d46e23f44368523535 Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Fri, 14 Feb 2025 11:47:00 +0800 Subject: [PATCH] =?UTF-8?q?refactor(llm):=20=E5=B0=86=E5=AD=A6=E4=B9=A0?= =?UTF-8?q?=E7=8E=87=E5=AD=97=E6=AE=B5=E7=B1=BB=E5=9E=8B=E4=BB=8E=20Intege?= =?UTF-8?q?r=20=E6=94=B9=E4=B8=BA=20BigDecimal?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 修改了 FineTuningTaskPageReqVO、FineTuningTaskRespVO 和 FineTuningTaskSaveReqVO 中的 learningRate 字段类型 - 优化了学习率的表示精度，提高了数据准确性 --- .../admin/finetuningtask/vo/FineTuningTaskPageReqVO.java | 6 ++++-- .../admin/finetuningtask/vo/FineTuningTaskRespVO.java | 4 +++- .../admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java | 6 ++++-- .../yudao/module/llm/service/http/TrainHttpService.java | 1 + 4 files changed, 12 insertions(+), 5 deletions(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java index 370491db1..9c377f1ec 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java @@ -1,6 +1,8 @@ package cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo; import lombok.*; + +import java.math.BigDecimal; import java.util.*; import io.swagger.v3.oas.annotations.media.Schema; import cn.iocoder.yudao.framework.common.pojo.PageParam; @@ -43,7 +45,7 @@ public class FineTuningTaskPageReqVO extends PageParam { private Long dataset; @Schema(description = "学习率") - private Integer learningRate; + private BigDecimal learningRate; @Schema(description = "训练轮数") private Integer epoch; @@ -94,4 +96,4 @@ public class FineTuningTaskPageReqVO extends PageParam { @Schema(description ="基础模型服务Id") private Long baseModelId; -} \ No newline at end of file +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java index eb65f0402..93fa2ccc9 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java @@ -4,6 +4,8 @@ import cn.iocoder.yudao.module.llm.service.http.vo.AigcFineTuningDetailRespVO; import com.alibaba.fastjson.JSONObject; import io.swagger.v3.oas.annotations.media.Schema; import lombok.*; + +import java.math.BigDecimal; import java.util.*; import org.springframework.format.annotation.DateTimeFormat; import java.time.LocalDateTime; @@ -58,7 +60,7 @@ public class FineTuningTaskRespVO { @Schema(description = "学习率") @ExcelProperty("学习率") - private Integer learningRate; + private BigDecimal learningRate; @Schema(description = "训练轮数") @ExcelProperty("训练轮数") diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java index 4451f79f8..5bb1d506c 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java @@ -2,6 +2,8 @@ package cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo; import io.swagger.v3.oas.annotations.media.Schema; import lombok.*; + +import java.math.BigDecimal; import java.util.*; import javax.validation.constraints.*; @@ -41,7 +43,7 @@ public class FineTuningTaskSaveReqVO { private Long dataset; @Schema(description = "学习率") - private Integer learningRate; + private BigDecimal learningRate; @Schema(description = "训练轮数") private Integer epoch; @@ -89,4 +91,4 @@ public class FineTuningTaskSaveReqVO { @Schema(description = "基础模型服务Id") private Long baseModelId; -} \ No newline at end of file +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java index e868db056..c5b29eac0 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java @@ -154,6 +154,7 @@ public class TrainHttpService { // TODO: 在上个方法中已经将数据集的文件id赋予，调试时需要写死再放开 // String fileId = "6237ed4d-a046-479c-80d6-8579a0283994"; // req.setFileId(fileId); + log.info(" finetuningCreate:{}", JSON.toJSONString(req)); String res = HttpUtils.post(llmBackendProperties.getFinetuningCreate(), headers, JSON.toJSONString(req)); log.info(" finetuningCreate:{}", res); AigcRespVO aigcRespVO = JSON.parseObject(res, AigcRespVO.class);