"git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "9a1213e0e534a14b5f907ec4a0033bd9954034ef"
Unverified Commit 37b359b2 authored by Kadir Nar's avatar Kadir Nar Committed by GitHub
Browse files

The variable name has been updated. (#2970)

parent a9477bbd
......@@ -405,14 +405,14 @@ def main():
args = parse_args()
logging_dir = Path(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps,
mixed_precision=args.mixed_precision,
log_with="tensorboard",
logging_dir=logging_dir,
accelerator_project_config=accelerator_project_config,
project_config=project_config,
)
# Currently, it's not possible to do gradient accumulation when training two models with accelerate.accumulate
......
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