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

chore: 调整超参数

Browse Source
This commit is contained in:
unanmed 2026-01-20 23:57:05 +08:00
parent 05b3b7c171
commit f22afb5f72
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ginka/train_vae.py
View File

@ -63,7 +63,7 @@ disable_tqdm = not sys.stdout.isatty()
def gt_prob(epoch: int, max_epoch: int) -> float:
progress = epoch / max_epoch
return max(1.2 * progress - 0.2, 0)
return max(2 * progress - 1, 0)
def parse_arguments():
parser = argparse.ArgumentParser(description="training codes")
@ -89,7 +89,7 @@ def train():
dataloader = DataLoader(dataset, batch_size=BATCH_SIZE, shuffle=True)
dataloader_val = DataLoader(dataset_val, batch_size=BATCH_SIZE // 4, shuffle=True)
optimizer_ginka = optim.AdamW(vae.parameters(), lr=1e-4, weight_decay=1e-4)
optimizer_ginka = optim.AdamW(vae.parameters(), lr=3e-4, weight_decay=1e-4)
scheduler_ginka = optim.lr_scheduler.CosineAnnealingLR(optimizer_ginka, T_max=800, eta_min=1e-6)
criterion = VAELoss()