Act on deprecations in Accelerate no_trainer examples (#24053)
Act on deprecation
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user