simplify code

yikangshen · Aug 6, 2018 · 7c64238 · 7c64238
1 parent ba64f50
commit 7c64238
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test_phrase_grammar.py b/test_phrase_grammar.py
@@ -140,7 +140,7 @@ def test(model, corpus, cuda, prt=False):
     # Model parameters.
     parser.add_argument('--data', type=str, default='./data/ptb',
                         help='location of the data corpus')
-    parser.add_argument('--checkpoint', type=str, default='./model/52476941.pt',
+    parser.add_argument('--checkpoint', type=str, default='./model/PTB.pt',
                         help='model checkpoint to use')
     parser.add_argument('--seed', type=int, default=1111,
                         help='random seed')