Browse Source

修复联邦学习应用bug

pull/56/head
jagger 3 months ago
parent
commit
7eb492d238
2 changed files with 9 additions and 2 deletions
  1. +2
    -0
      sdks/scheduler/app.go
  2. +7
    -2
      sdks/scheduler/app_models.go

+ 2
- 0
sdks/scheduler/app.go View File

@@ -37,6 +37,7 @@ type AppInstanceDao struct {
AppRunRecords []AppRunRecordDao `gorm:"foreignKey:AppInstanceID;references:ID" json:"appRunRecords"` //运行记录
Status string `json:"status"` //应用状态
Role string `json:"role"` //应用角色
JobSetID JobSetID `json:"jobSetID"` // 关联的作业集ID
}

func (AppInstanceDao) TableName() string {
@@ -59,6 +60,7 @@ type AppInstance struct {
AppRunRecords []AppRunRecordDao `gorm:"foreignKey:AppInstanceID;references:ID" json:"appRunRecords"` //运行记录
Status string `json:"status"` //应用状态
Role string `gorm:"-" json:"role"` //应用角色
JobSetID JobSetID `gorm:"-" json:"jobSetID"` // 关联的作业集ID
}

type AppRunRecordDao struct {


+ 7
- 2
sdks/scheduler/app_models.go View File

@@ -55,13 +55,18 @@ type FederatedLearningAppInfo struct {
Name string `json:"name"`
Description string `json:"description"`
ClientCount int `json:"clientCount"` //参与者人数
ServerJobInfo JobInfo `json:"serverJobInfo"` //服务端作业信息
ClientJobInfo JobInfo `json:"clientJobInfo"` //客户端作业信息
ServerJobInfo AIJobInfo `json:"serverJobInfo"` //服务端作业信息
ClientJobInfo AIJobInfo `json:"clientJobInfo"` //客户端作业信息
Role string `json:"role"` //角色,server/client
IsDataReturn bool `json:"isDataReturn"` //是否需要数据回源
DatasetBindID cdssdk.ObjectID `json:"datasetBindID"` //数据集绑定ID
ClientIDs []ClientID `json:"clientIDs"` //参与者ID列表
JoinedCount int64 `json:"joinedCount"` //已经参与的人数

//用于回显数据
ClientDatasets []interface{} `json:"clientDatasets"` //数据集列表
ServerDataDistribute interface{} `json:"serverDataDistribute"` //server数据分发信息
ClientDataDistribute interface{} `json:"clientDataDistribute"` //client数据分发信息
}

type OtherAppInfo struct {


Loading…
Cancel
Save