修改配置

This commit is contained in:
limin 2025-02-14 10:05:43 +08:00
parent 1f20747f9c
commit 99685d3605

View File

@ -74,7 +74,7 @@ public class AsyncFineTuningTaskService {
req.setSuffix(fineTuningTask.getTaskName());
req.setRemark(fineTuningTask.getTaskIntro());
req.setTrainBatchSize(fineTuningTask.getBatchSize());
req.setEvalBatchSize(fineTuningTask.getBatchSize());
req.setEvalBatchSize(32);
req.setAccumulationSteps(fineTuningTask.getGradientAccumulation());
req.setProcPerNode(fineTuningTask.getGpuCount());
req.setLearningRate(fineTuningTask.getLearningRate());