SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Simplify code, delete redundancy line

Browse Source 
delete redundancy line `if args.train`, simplify code.
This commit is contained in:
Yongbo Wang
2019-03-13 09:42:06 +08:00
committed by GitHub
parent 9b03d67b83
commit 22a465a91f
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/run_classifier.py
View File

@@ -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)