From a6a16895d6bc52f41f3d5da5dc7b2e3b7de74400 Mon Sep 17 00:00:00 2001 From: mehrad Date: Mon, 27 Apr 2020 21:00:10 -0700 Subject: [PATCH] update ckpt default value --- .gitignore | 5 +++-- genienlp/paraphrase/evaluate_bart.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/.gitignore b/.gitignore index a5e9fe66..97cdb5f6 100644 --- a/.gitignore +++ b/.gitignore @@ -23,14 +23,15 @@ src/ workdir/ *save*/ test_py.sh +/tests/*/ + # paraphraser */paraphrase/*.txt +*/paraphrase/bart-* lightning_logs/ pytorch_model.bin -*/paraphrase/bart-* cnn* -/tests/*/ # C extensions diff --git a/genienlp/paraphrase/evaluate_bart.py b/genienlp/paraphrase/evaluate_bart.py index 10914a21..bb1a69d2 100644 --- a/genienlp/paraphrase/evaluate_bart.py +++ b/genienlp/paraphrase/evaluate_bart.py @@ -63,7 +63,7 @@ if __name__ == "__main__": parser.add_argument("--device", type=str, required=False, default=DEFAULT_DEVICE, help="cuda, cuda:1, cpu etc.") parser.add_argument("--batch_size", type=int, default=8, required=False, help="batch size: how many to summarize at a time") parser.add_argument("--ckpt_path", type=str, required=True, help="path to checkpoint file") - parser.add_argument("--ckpt_name", type=str, default='checkpointepoch=0.ckpt', help="checkpoint name") + parser.add_argument("--ckpt_name", type=str, default='mbart-epoch=00.ckpt', help="checkpoint name") parser.add_argument("--predict_split", type=str, default='test', help="data split to run prediction for") parser.add_argument( "--max_source_length",