From f7c9c424e4249d1d395fce0b6eb8914325f60352 Mon Sep 17 00:00:00 2001 From: Behzad Shayegh <43534816+ShayeghB@users.noreply.github.com> Date: Sun, 17 Sep 2023 11:32:25 -0600 Subject: [PATCH] Update README.md (epochs -> num_epochs) --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 1223761..4109686 100644 --- a/README.md +++ b/README.md @@ -125,7 +125,7 @@ For fine-tuning: python train.py --train_from compound-pcfg-rnng.pt --save_path compound-pcfg-urnng.pt --train_file /compound-pcfg/data/ptb-comp-pcfg-train.pkl --val_file /compound-pcfg/data/ptb-comp-pcfg-val.pkl ---mode unsupervised --train_q_epochs 10 --epochs 10 --count_eos_ppl 1 --lr 0.1 --gpu 0 --kl_warmup 0 +--mode unsupervised --train_q_epochs 10 --num_epochs 10 --count_eos_ppl 1 --lr 0.1 --gpu 0 --kl_warmup 0 ``` For evaluation: