"references/vscode:/vscode.git/clone" did not exist on "5bb81c8e008c601237e2ffd5cbea7192775130bd"
Unverified Commit 46def726 authored by Will Berman's avatar Will Berman Committed by GitHub
Browse files

checkpointing_steps_total_limit->checkpoints_total_limit (#2374)

parent 296b01e1
...@@ -196,7 +196,7 @@ def parse_args(input_args=None): ...@@ -196,7 +196,7 @@ def parse_args(input_args=None):
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -498,7 +498,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: ...@@ -498,7 +498,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def main(args): def main(args):
logging_dir = Path(args.output_dir, args.logging_dir) logging_dir = Path(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -243,7 +243,7 @@ def parse_args(input_args=None): ...@@ -243,7 +243,7 @@ def parse_args(input_args=None):
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -536,7 +536,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: ...@@ -536,7 +536,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def main(args): def main(args):
logging_dir = Path(args.output_dir, args.logging_dir) logging_dir = Path(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -259,7 +259,7 @@ def parse_args(): ...@@ -259,7 +259,7 @@ def parse_args():
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -416,7 +416,7 @@ def main(): ...@@ -416,7 +416,7 @@ def main():
args = parse_args() args = parse_args()
logging_dir = Path(args.output_dir, args.logging_dir) logging_dir = Path(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -255,7 +255,7 @@ def parse_args(): ...@@ -255,7 +255,7 @@ def parse_args():
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -415,7 +415,7 @@ def main(): ...@@ -415,7 +415,7 @@ def main():
args = parse_args() args = parse_args()
logging_dir = Path(args.output_dir, args.logging_dir) logging_dir = Path(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -171,7 +171,7 @@ def parse_args(input_args=None): ...@@ -171,7 +171,7 @@ def parse_args(input_args=None):
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -476,7 +476,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: ...@@ -476,7 +476,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def main(args): def main(args):
logging_dir = Path(args.output_dir, args.logging_dir) logging_dir = Path(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -275,7 +275,7 @@ def parse_args(): ...@@ -275,7 +275,7 @@ def parse_args():
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -332,7 +332,7 @@ def main(): ...@@ -332,7 +332,7 @@ def main():
args = parse_args() args = parse_args()
logging_dir = os.path.join(args.output_dir, args.logging_dir) logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -291,7 +291,7 @@ def parse_args(): ...@@ -291,7 +291,7 @@ def parse_args():
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -477,7 +477,7 @@ def main(): ...@@ -477,7 +477,7 @@ def main():
args = parse_args() args = parse_args()
logging_dir = os.path.join(args.output_dir, args.logging_dir) logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -233,7 +233,7 @@ def parse_args(): ...@@ -233,7 +233,7 @@ def parse_args():
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -276,7 +276,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: ...@@ -276,7 +276,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def main(args): def main(args):
logging_dir = os.path.join(args.output_dir, args.logging_dir) logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -276,7 +276,7 @@ def parse_args(): ...@@ -276,7 +276,7 @@ def parse_args():
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -343,7 +343,7 @@ def main(): ...@@ -343,7 +343,7 @@ def main():
) )
logging_dir = os.path.join(args.output_dir, args.logging_dir) logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -311,7 +311,7 @@ def parse_args(): ...@@ -311,7 +311,7 @@ def parse_args():
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -364,7 +364,7 @@ def main(): ...@@ -364,7 +364,7 @@ def main():
args = parse_args() args = parse_args()
logging_dir = os.path.join(args.output_dir, args.logging_dir) logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -289,7 +289,7 @@ def parse_args(): ...@@ -289,7 +289,7 @@ def parse_args():
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -475,7 +475,7 @@ def main(): ...@@ -475,7 +475,7 @@ def main():
args = parse_args() args = parse_args()
logging_dir = os.path.join(args.output_dir, args.logging_dir) logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
...@@ -241,7 +241,7 @@ def parse_args(): ...@@ -241,7 +241,7 @@ def parse_args():
), ),
) )
parser.add_argument( parser.add_argument(
"--checkpointing_steps_total_limit", "--checkpoints_total_limit",
type=int, type=int,
default=None, default=None,
help=( help=(
...@@ -284,7 +284,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: ...@@ -284,7 +284,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def main(args): def main(args):
logging_dir = os.path.join(args.output_dir, args.logging_dir) logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment