1
This commit is contained in:
@@ -26,7 +26,6 @@ import (
|
||||
// tasks/<taskId>.json 任务参数(Go 侧写入)
|
||||
// logs/<taskId>.jsonl 每 epoch 一行 JSON:{"epoch","total","metrics"}
|
||||
// results/<taskId>.json 结束结果:{"metrics","names","best_tflite"(相对路径)}
|
||||
// artifacts/<taskId>.zip 打包产物(best.pt + results.csv + 曲线)
|
||||
type TrainingRunner interface {
|
||||
// Start 启动训练进程,返回可探测存活的 pid(subprocess 本机 pid;ssh 远程 pid)
|
||||
Start(ctx context.Context, job *TrainingJob) (int, error)
|
||||
|
||||
@@ -5,7 +5,6 @@ import (
|
||||
"crypto/sha256"
|
||||
"encoding/hex"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
)
|
||||
@@ -13,8 +12,7 @@ import (
|
||||
// 训练体系运行时数据布局(config.yml app.datasetDir,默认 ./workspace,挂载持久化、不提交 git):
|
||||
//
|
||||
// datasets/<name>/ 数据集图片(平铺,文件名唯一,标注存 DB dataset_image.labels_json)
|
||||
// models/<name>/ latest.tflite 当前生效副本(无存档,客户端固定下载)
|
||||
// trainings/<taskId>/ 训练产物(best.tflite + artifact.zip)
|
||||
// trainings/<name>.tflite 某数据集当前生效模型(训练成功即直写,客户端固定下载,无存档回退)
|
||||
//
|
||||
// 训练机与 Go 服务器异机时,数据集经 training 通道同步(见 common/training_runner.go)。
|
||||
|
||||
@@ -28,14 +26,9 @@ func DatasetImagesDir(ctx context.Context, datasetName string) string {
|
||||
return filepath.Join(DatasetDir(ctx), "datasets", datasetName)
|
||||
}
|
||||
|
||||
// DatasetModelsDir 某数据集模型目录
|
||||
func DatasetModelsDir(ctx context.Context, datasetName string) string {
|
||||
return filepath.Join(DatasetDir(ctx), "models", datasetName)
|
||||
}
|
||||
|
||||
// TrainingArtifactsDir 某训练任务产物目录(拉回的 best.tflite + artifact.zip)
|
||||
func TrainingArtifactsDir(ctx context.Context, taskId int64) string {
|
||||
return filepath.Join(DatasetDir(ctx), "trainings", strconv.FormatInt(taskId, 10))
|
||||
// TrainingModelPath 某数据集当前生效模型文件(训练成功即从训练机直写,客户端固定下载)
|
||||
func TrainingModelPath(ctx context.Context, datasetName string) string {
|
||||
return filepath.Join(DatasetDir(ctx), "trainings", datasetName+".tflite")
|
||||
}
|
||||
|
||||
// Sha256Hex 计算文件内容 SHA-256 十六进制(模型版本校验用)
|
||||
|
||||
Reference in New Issue
Block a user