Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
c64de50e
Commit
c64de50e
authored
Jan 03, 2019
by
Jade Abbott
Browse files
nb_tr_steps is not initialized
parent
b96149a1
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
0 deletions
+2
-0
examples/run_classifier.py
examples/run_classifier.py
+2
-0
No files found.
examples/run_classifier.py
View file @
c64de50e
...
...
@@ -503,6 +503,7 @@ def main():
t_total
=
t_total
)
global_step
=
0
nb_tr_steps
=
0
tr_loss
=
0
if
args
.
do_train
:
train_features
=
convert_examples_to_features
(
...
...
@@ -565,6 +566,7 @@ def main():
if
args
.
do_eval
and
(
args
.
local_rank
==
-
1
or
torch
.
distributed
.
get_rank
()
==
0
):
eval_examples
=
processor
.
get_dev_examples
(
args
.
data_dir
)
# should tokenize this too.
eval_features
=
convert_examples_to_features
(
eval_examples
,
label_list
,
args
.
max_seq_length
,
tokenizer
)
logger
.
info
(
"***** Running evaluation *****"
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment