refactor(llm): 重构模型微调相关代码
- 将 AigcFineTuningCreateReqVO 和 FineTuningTaskDO 中的 learningRate 字段类型从 double/Integer 改为 BigDecimal,以提高精度 - 优化 ModelService 中的 createFineTuningTask 方法,异常处理改为抛出 RuntimeException
This commit is contained in:
parent
08c82dbb83
commit
1f20747f9c
@ -1,6 +1,8 @@
|
||||
package cn.iocoder.yudao.module.llm.dal.dataobject.finetuningtask;
|
||||
|
||||
import lombok.*;
|
||||
|
||||
import java.math.BigDecimal;
|
||||
import java.util.*;
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.LocalDateTime;
|
||||
@ -71,7 +73,7 @@ public class FineTuningTaskDO extends BaseDO {
|
||||
/**
|
||||
* 学习率
|
||||
*/
|
||||
private Integer learningRate;
|
||||
private BigDecimal learningRate;
|
||||
/**
|
||||
* 训练轮数
|
||||
*/
|
||||
|
@ -90,7 +90,7 @@ public class ModelService {
|
||||
}
|
||||
return null;
|
||||
} catch (Exception e) {
|
||||
return null;
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -4,6 +4,8 @@ import lombok.AllArgsConstructor;
|
||||
import lombok.Data;
|
||||
import lombok.NoArgsConstructor;
|
||||
|
||||
import java.math.BigDecimal;
|
||||
|
||||
@Data
|
||||
@AllArgsConstructor
|
||||
@NoArgsConstructor
|
||||
@ -20,7 +22,7 @@ public class AigcFineTuningCreateReqVO {
|
||||
private int evalBatchSize;
|
||||
private int accumulationSteps;
|
||||
private int procPerNode;
|
||||
private double learningRate;
|
||||
private BigDecimal learningRate;
|
||||
private int modelMaxLength;
|
||||
private boolean lora;
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user