saily commited on
Commit
0fd83b3
·
1 Parent(s): 5a20515

change pretest

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -17,7 +17,7 @@ parser.add_argument("--mode", type=str, required=True, help="train/demo/predict"
17
  parser.add_argument("--data_dir", type=str, default="./data", help="training data and saved model path")
18
  #parser.add_argument("--pretrained_bert_dir", type=str, default="./pretrained_bert", help="pretrained bert model path")
19
  parser.add_argument("--seed", type=int, default=1, help="random seed for initialization")
20
- parser.add_argument("--input_file", type=str, default="./data/input.txt", help="input file to be predicted")
21
  args = parser.parse_args()
22
 
23
  args.pretrained_bert_dir = "bert-base-chinese"
 
17
  parser.add_argument("--data_dir", type=str, default="./data", help="training data and saved model path")
18
  #parser.add_argument("--pretrained_bert_dir", type=str, default="./pretrained_bert", help="pretrained bert model path")
19
  parser.add_argument("--seed", type=int, default=1, help="random seed for initialization")
20
+ parser.add_argument("--input_file", type=str, default="./data_12345/test.txt", help="input file to be predicted")
21
  args = parser.parse_args()
22
 
23
  args.pretrained_bert_dir = "bert-base-chinese"