diff --git a/src/pages/api/model/train.ts b/src/pages/api/model/train.ts index b63dd85a3..9efe71341 100644 --- a/src/pages/api/model/train.ts +++ b/src/pages/api/model/train.ts @@ -84,7 +84,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse) training_file: uploadFileId, model: trainingType, suffix: model.name, - n_epochs: 2 + n_epochs: 4 }, { httpsAgent } ); diff --git a/src/pages/model/detail.tsx b/src/pages/model/detail.tsx index 50c0e44e7..4c1063b24 100644 --- a/src/pages/model/detail.tsx +++ b/src/pages/model/detail.tsx @@ -102,7 +102,7 @@ const ModelDetail = ({ modelId }: { modelId: string }) => { await postTrainModel(modelId, formData); toast({ - title: '开始训练,大约需要 30 分钟', + title: '开始训练...', status: 'success' });