Skip to content

Commit 640f984

Browse files
committed
quick fix error evaluation during training
1 parent 9081abc commit 640f984

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

cdqa/reader/reader_sklearn.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -388,8 +388,6 @@ def __init__(self,
388388
max_seq_length=384,
389389
doc_stride=128,
390390
max_query_length=64,
391-
do_train=True,
392-
do_eval=True,
393391
evaluate_during_training=True,
394392
do_lower_case=True,
395393
per_gpu_train_batch_size=8,

examples/tutorial-train-xlnet-squad.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,10 @@
1616
# cast Reader class with train params
1717
reader = Reader(model_type='xlnet',
1818
model_name_or_path='xlnet-base-cased',
19-
fp16=False,
20-
output_dir='.')
19+
output_dir='.',
20+
evaluate_during_training=False,
21+
no_cuda=True,
22+
fp16=False)
2123

2224
# train the model
2325
reader.fit(X='train-v2.0.json')

0 commit comments

Comments
 (0)