训练任务接口

提交训练任务

POST /api/training/tasks

请求体

{
  "project_id": 1,
  "workflow_id": 1,
  "params": {
    "epochs": 100,
    "batch_size": 8,
    "learning_rate": 0.0001,
    "optimizer": "adamw",
    "loss_function": "dice_loss"
  },
  "worker_type": "gpu"
}

响应

{
  "task_id": "550e8400-e29b-41d4-a716-446655440000",
  "status": "queued",
  "created_at": "2026-01-01T00:00:00Z"
}

查询任务状态

GET /api/training/tasks/{task_id}

响应

{
  "task_id": "550e8400-e29b-41d4-a716-446655440000",
  "status": "running",
  "progress": {
    "current_epoch": 50,
    "total_epochs": 100,
    "current_loss": 0.1234,
    "best_dice": 0.8765
  },
  "started_at": "2026-01-01T00:01:00Z"
}

任务状态

状态	说明
queued	等待执行
running	执行中
completed	已完成
failed	执行失败
cancelled	已取消

查询任务列表

GET /api/training/tasks?project_id=1&status=completed

取消任务

POST /api/training/tasks/{task_id}/cancel

获取训练结果

GET /api/training/tasks/{task_id}/results

响应

{
  "metrics": {
    "dice": 0.8765,
    "iou": 0.7890,
    "hausdorff": 2.34,
    "sensitivity": 0.9012,
    "specificity": 0.9876
  },
  "model_path": "models/project_1/best_model.pth",
  "training_time": 3600
}

获取训练日志

GET /api/training/tasks/{task_id}/logs