AncTe/ginka-generator
1
0
Fork 0
mirror of https://github.com/unanmed/ginka-generator.git synced 2026-05-20 00:51:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: 调整调度

Browse Source
This commit is contained in:
unanmed 2026-02-13 00:05:38 +08:00
parent becf625bdb
commit c8809b8ee7
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ginka/train_vae.py
View File

@ -58,7 +58,8 @@ LATENT_DIM = 48
KL_BETA = 0.1 KL_BETA = 0.1
SELF_GATE = 0.5 SELF_GATE = 0.5
GATE_EPOCH = 5 GATE_EPOCH = 5
VAL_BATCH_DIVIDER = 1 VAL_BATCH_DIVIDER = 128
PROB_STEP = 0.05
device = torch.device( device = torch.device(
"cuda:1" if torch.cuda.is_available() "cuda:1" if torch.cuda.is_available()
@ -168,10 +169,14 @@ def train():
# 先使用训练集的损失值，因为过拟合比较严重，后续再想办法 # 先使用训练集的损失值，因为过拟合比较严重，后续再想办法
if avg_loss < SELF_GATE: if avg_loss < SELF_GATE:
prob_epochs += 1 prob_epochs += 1
else:
prob_epochs = 0
if prob_epochs >= GATE_EPOCH and self_prob < 1: if prob_epochs >= GATE_EPOCH and self_prob < 1:
self_prob += 0.01 self_prob += PROB_STEP
prob_epochs = 0 prob_epochs = 0
if self_prob > 1:
self_prob = 1
scheduler_ginka.step(avg_loss, self_prob) scheduler_ginka.step(avg_loss, self_prob)