"vscode:/vscode.git/clone" did not exist on "6c2ee16c0418a09c13cd59bf285f56feb001d3b5"
Unverified Commit e99071f1 authored by Thomas Wolf's avatar Thomas Wolf Committed by GitHub
Browse files

Merge pull request #1734 from orena1/patch-1

add progress bar to convert_examples_to_features
parents ba973342 d7906165
...@@ -23,6 +23,7 @@ import logging ...@@ -23,6 +23,7 @@ import logging
import math import math
import collections import collections
from io import open from io import open
from tqdm import tqdm
from transformers.tokenization_bert import BasicTokenizer, whitespace_tokenize from transformers.tokenization_bert import BasicTokenizer, whitespace_tokenize
...@@ -202,7 +203,7 @@ def convert_examples_to_features(examples, tokenizer, max_seq_length, ...@@ -202,7 +203,7 @@ def convert_examples_to_features(examples, tokenizer, max_seq_length,
# f = np.zeros((max_N, max_M), dtype=np.float32) # f = np.zeros((max_N, max_M), dtype=np.float32)
features = [] features = []
for (example_index, example) in enumerate(examples): for (example_index, example) in enumerate(tqdm(examples)):
# if example_index % 100 == 0: # if example_index % 100 == 0:
# logger.info('Converting %s/%s pos %s neg %s', example_index, len(examples), cnt_pos, cnt_neg) # logger.info('Converting %s/%s pos %s neg %s', example_index, len(examples), cnt_pos, cnt_neg)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment