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

examples/run_summarization_no_trainer: fixed incorrect param to hasattr (#18720)

Browse Source 
* fixed incorrect param to hasattr

* simplified condition checks

* code cleanup
This commit is contained in:
Rahul A R
2022-08-24 12:12:42 -04:00
committed by GitHub
parent 6667b0d7bf
commit c55d6e4e10
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/pytorch/summarization/run_summarization_no_trainer.py
View File

@@ -573,12 +573,9 @@ def main():
args.num_train_epochs = math.ceil(args.max_train_steps / num_update_steps_per_epoch)
# Figure out how many steps we should save the Accelerator states
if hasattr(args.checkpointing_steps, "isdigit"):
checkpointing_steps = args.checkpointing_steps
if args.checkpointing_steps.isdigit():
checkpointing_steps = int(args.checkpointing_steps)
else:
checkpointing_steps = None
if checkpointing_steps is not None and checkpointing_steps.isdigit():
checkpointing_steps = int(checkpointing_steps)
# We need to initialize the trackers we use, and also store our configuration.
# The trackers initializes automatically on the main process.