riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 13:20:12 +06:00
Code Issues Actions 21 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 6 deletions
Show Stats 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) 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 # Figure out how many steps we should save the Accelerator states
if hasattr(args.checkpointing_steps, "isdigit"):
checkpointing_steps = args.checkpointing_steps checkpointing_steps = args.checkpointing_steps
if args.checkpointing_steps.isdigit(): if checkpointing_steps is not None and checkpointing_steps.isdigit():
checkpointing_steps = int(args.checkpointing_steps) checkpointing_steps = int(checkpointing_steps)
else:
checkpointing_steps = None
# We need to initialize the trackers we use, and also store our configuration. # We need to initialize the trackers we use, and also store our configuration.
# The trackers initializes automatically on the main process. # The trackers initializes automatically on the main process.