change pretest
Browse files
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="./
|
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"
|