Skip to content

GitLab

Unverified Commit eecaaa73 authored by Thomas Wolf's avatar Thomas Wolf Committed by GitHub
Browse files

Merge pull request #371 from yongbowin/patch-1 

Simplify code, delete redundancy line
parents 9b03d67b 22a465a9
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
examples/run_classifier.py
View file @ eecaaa73
...@@ -594,7 +594,6 @@ def main(): ...@@ -594,7 +594,6 @@ def main():
optimizer.zero_grad() optimizer.zero_grad()
global_step += 1 global_step += 1
if args.do_train:
# Save a trained model and the associated configuration # Save a trained model and the associated configuration
model_to_save = model.module if hasattr(model, 'module') else model # Only save the model it-self model_to_save = model.module if hasattr(model, 'module') else model # Only save the model it-self
output_model_file = os.path.join(args.output_dir, WEIGHTS_NAME) output_model_file = os.path.join(args.output_dir, WEIGHTS_NAME)
......
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