@daocloud-proto/hydra 0.0.0-dev-cb927b7b → 0.0.0-dev-d8ebaf68
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
|
@@ -102,25 +102,26 @@ export type DatasetConfig = {
|
|
|
102
102
|
datasetVolumeId?: string
|
|
103
103
|
cutoffLen?: number
|
|
104
104
|
maxSamples?: number
|
|
105
|
-
valSize?:
|
|
105
|
+
valSize?: number
|
|
106
106
|
datasetName?: string
|
|
107
|
+
datasetTag?: string
|
|
107
108
|
}
|
|
108
109
|
|
|
109
110
|
export type TrainingConfig = {
|
|
110
|
-
numTrainEpochs?:
|
|
111
|
+
numTrainEpochs?: number
|
|
111
112
|
perDeviceTrainBatchSize?: number
|
|
112
113
|
gradientAccumulationSteps?: number
|
|
113
|
-
learningRate?:
|
|
114
|
+
learningRate?: number
|
|
114
115
|
lrSchedulerType?: LrSchedulerType
|
|
115
|
-
warmupRatio?:
|
|
116
|
+
warmupRatio?: number
|
|
116
117
|
warmupSteps?: number
|
|
117
118
|
computeType?: ComputeType
|
|
118
119
|
optimizer?: OptimizerType
|
|
119
120
|
seed?: number
|
|
120
121
|
ddpTimeout?: number
|
|
121
|
-
maxGradNorm?:
|
|
122
|
+
maxGradNorm?: number
|
|
122
123
|
neftAlpha?: number
|
|
123
|
-
deepspeedStage?:
|
|
124
|
+
deepspeedStage?: number
|
|
124
125
|
deepspeedOffload?: boolean
|
|
125
126
|
gradientCheckpointing?: boolean
|
|
126
127
|
loggingSteps?: number
|
|
@@ -129,10 +130,10 @@ export type TrainingConfig = {
|
|
|
129
130
|
}
|
|
130
131
|
|
|
131
132
|
export type DpoConfig = {
|
|
132
|
-
prefBeta?:
|
|
133
|
+
prefBeta?: number
|
|
133
134
|
prefLoss?: string
|
|
134
|
-
prefFtx?:
|
|
135
|
-
dpoLabelSmoothing?:
|
|
135
|
+
prefFtx?: number
|
|
136
|
+
dpoLabelSmoothing?: number
|
|
136
137
|
refModel?: string
|
|
137
138
|
}
|
|
138
139
|
|
|
@@ -163,6 +164,7 @@ export type FineTuneConfig = {
|
|
|
163
164
|
createdAt?: GoogleProtobufTimestamp.Timestamp
|
|
164
165
|
updatedAt?: GoogleProtobufTimestamp.Timestamp
|
|
165
166
|
modelSource?: boolean
|
|
167
|
+
workspace?: number
|
|
166
168
|
}
|
|
167
169
|
|
|
168
170
|
export type CreateWSFineTuneConfigRequest = {
|