update ckpt default value

This commit is contained in:
mehrad 2020-04-27 21:00:10 -07:00
parent 9dc73b5ce4
commit a6a16895d6
2 changed files with 4 additions and 3 deletions

5
.gitignore vendored
View File

@ -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

View File

@ -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",