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

streamlining 'checkpointing_steps' parsing (#18755)

Browse Source
This commit is contained in:
Rahul A R
2022-08-25 11:00:38 -04:00
committed by GitHub
parent fbf382c84d
commit e9442440fc
10 changed files with 30 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/pytorch/text-classification/run_glue_no_trainer.py
View File

@@ -451,12 +451,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
checkpointing_steps = args.checkpointing_steps
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.