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):
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -498,7 +498,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def main(args):
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -243,7 +243,7 @@ def parse_args(input_args=None):
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -536,7 +536,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def main(args):
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -259,7 +259,7 @@ def parse_args():
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -416,7 +416,7 @@ def main():
args = parse_args()
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -255,7 +255,7 @@ def parse_args():
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -415,7 +415,7 @@ def main():
args = parse_args()
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -171,7 +171,7 @@ def parse_args(input_args=None):
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -476,7 +476,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def main(args):
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -275,7 +275,7 @@ def parse_args():
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -332,7 +332,7 @@ def main():
args = parse_args()
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -291,7 +291,7 @@ def parse_args():
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -477,7 +477,7 @@ def main():
args = parse_args()
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -233,7 +233,7 @@ def parse_args():
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -276,7 +276,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def main(args):
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -276,7 +276,7 @@ def parse_args():
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -343,7 +343,7 @@ def main():
)
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -311,7 +311,7 @@ def parse_args():
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -364,7 +364,7 @@ def main():
args = parse_args()
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -289,7 +289,7 @@ def parse_args():
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -475,7 +475,7 @@ def main():
args = parse_args()
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(
gradient_accumulation_steps=args.gradient_accumulation_steps,
......
......@@ -241,7 +241,7 @@ def parse_args():
),
)
parser.add_argument(
"--checkpointing_steps_total_limit",
"--checkpoints_total_limit",
type=int,
default=None,
help=(
......@@ -284,7 +284,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def main(args):
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(
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