Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "65be574aeca3e8c4bfb25065ea98deed7f3cc471"
Commit c64de50e authored by Jade Abbott's avatar Jade Abbott
Browse files

nb_tr_steps is not initialized

parent b96149a1
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
examples/run_classifier.py
View file @ c64de50e
...@@ -503,6 +503,7 @@ def main(): ...@@ -503,6 +503,7 @@ def main():
t_total=t_total) t_total=t_total)
global_step = 0 global_step = 0
nb_tr_steps = 0
tr_loss = 0 tr_loss = 0
if args.do_train: if args.do_train:
train_features = convert_examples_to_features( train_features = convert_examples_to_features(
...@@ -565,6 +566,7 @@ def main(): ...@@ -565,6 +566,7 @@ def main():
if args.do_eval and (args.local_rank == -1 or torch.distributed.get_rank() == 0): if args.do_eval and (args.local_rank == -1 or torch.distributed.get_rank() == 0):
eval_examples = processor.get_dev_examples(args.data_dir) eval_examples = processor.get_dev_examples(args.data_dir)
# should tokenize this too.
eval_features = convert_examples_to_features( eval_features = convert_examples_to_features(
eval_examples, label_list, args.max_seq_length, tokenizer) eval_examples, label_list, args.max_seq_length, tokenizer)
logger.info("***** Running evaluation *****") logger.info("***** Running evaluation *****")
......
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