DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

修改saveSteps

Browse Source
This commit is contained in:
leon 2025-02-28 15:18:53 +08:00
parent 7bdba8713b
commit 337237ee66
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
View File

@ -177,9 +177,9 @@ public class AsyncFineTuningTaskService {
AigcFineTuningCreateReqVO req = new AigcFineTuningCreateReqVO();
req.setModel(fineTuningTask.getBaseModel());
req.setTrainEpoch((fineTuningTask.getEpoch() == 0) ? 1 : fineTuningTask.getEpoch());
int saveStep = Optional.ofNullable(fineTuningTask.getSaveSteps())
int saveSteps = Optional.ofNullable(fineTuningTask.getSaveSteps())
.orElse(0) + 1;
req.setTrainEpoch(saveStep);
req.setSaveSteps(saveSteps);
req.setSuffix(fineTuningTask.getTaskName());
req.setRemark(fineTuningTask.getTaskIntro());
req.setTrainBatchSize(fineTuningTask.getBatchSize());

1
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/AigcFineTuningCreateReqVO.java
View File

@ -16,6 +16,7 @@ public class AigcFineTuningCreateReqVO {
private String dataset;
private String model;
private int trainEpoch;
private int saveSteps;
private String suffix;
private String remark;
private int trainBatchSize;