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/semantic-segmentation/run_semantic_segmentation_no_trainer.py
View File

@@ -475,12 +475,9 @@ def main():
)
# 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)
# Scheduler and math around the number of training steps.
overrode_max_train_steps = False