Act on deprecations in Accelerate no_trainer examples (#24053)

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

View File

@@ -7,6 +7,7 @@ from pathlib import Path
import datasets
import torch
from accelerate import Accelerator, DistributedType
from accelerate.utils import ProjectConfiguration
from arguments import TrainingArguments
from datasets import load_dataset
from huggingface_hub import Repository
@@ -195,7 +196,8 @@ parser = HfArgumentParser(TrainingArguments)
args = parser.parse_args()
# Accelerator
accelerator = Accelerator(log_with=["wandb", "tensorboard"], logging_dir=f"{args.save_dir}/log")
config = ProjectConfiguration(project_dir=args.save_dir, logging_dir="log")
accelerator = Accelerator(log_with=["wandb", "tensorboard"], project_config=config)
acc_state = {str(k): str(v) for k, v in accelerator.state.__dict__.items()}
args = Namespace(**vars(args), **acc_state)