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

fix: 数据集路径

Browse Source
This commit is contained in:
unanmed 2026-04-26 23:49:17 +08:00
parent 068940cae0
commit cc463f543e
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_vq.py
View File

@ -89,8 +89,8 @@ def parse_arguments():
parser.add_argument("--resume", type=bool, default=False)
parser.add_argument("--state", type=str, default="result/joint/joint-10.pth",
help="续训时加载的检查点路径")
parser.add_argument("--train", type=str, default="data/ginka-dataset.json")
parser.add_argument("--validate", type=str, default="data/ginka-eval.json")
parser.add_argument("--train", type=str, default="ginka-dataset.json")
parser.add_argument("--validate", type=str, default="ginka-eval.json")
parser.add_argument("--epochs", type=int, default=100)
parser.add_argument("--checkpoint", type=int, default=5,
help="每隔多少 epoch 保存检查点并验证")