This commit is contained in:
2026-02-03 10:21:07 +08:00
parent e556f17ebc
commit 0c2ae25cf8
81 changed files with 572 additions and 76 deletions

View File

@@ -15,7 +15,7 @@ INPUT:
OPTIM:
NAME: "sgd"
LR: 0.0035
MAX_EPOCH: 5
MAX_EPOCH: 10
LR_SCHEDULER: "cosine"
WARMUP_EPOCH: 1
WARMUP_TYPE: "constant"
@@ -38,4 +38,4 @@ TRAINER:
N_CTX: 2
CTX_INIT: "a photo of a"
PREC: "fp16"
PROMPT_DEPTH: 9
PROMPT_DEPTH: 9