修改配置
This commit is contained in:
parent
1f20747f9c
commit
99685d3605
@ -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());
|
||||
|
Loading…
x
Reference in New Issue
Block a user