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

Act on deprecations in Accelerate no_trainer examples (#24053)

Browse Source 
Act on deprecation
This commit is contained in:
Zachary Mueller
2023-06-06 13:04:38 -04:00
committed by GitHub
parent ff4c0fc7d2
commit 072188d638
12 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/semantic-segmentation/run_semantic_segmentation_no_trainer.py
View File

@@ -330,7 +330,7 @@ def main():
if args.with_tracking:
accelerator_log_kwargs["log_with"] = args.report_to
accelerator_log_kwargs["logging_dir"] = args.output_dir
accelerator_log_kwargs["project_dir"] = args.output_dir
accelerator = Accelerator(gradient_accumulation_steps=args.gradient_accumulation_steps, **accelerator_log_kwargs)