TypeError: Accelerator.init() got an unexpected keyword argument ‘logging_dir’

在使用 accelerate 0.20.3版本时,报TypeError: Accelerator.__init__() got an unexpected keyword argument 'logging_dir' 错误。
BUG排查发现,在 accelerate 0.18.x版本之后logging_dir这个参数已经被弃用并且被删除,现在采用的是 project_dir 参数,因此有两个解决思路:
(1)修改代码将Accelerator初始化时的logging_dir修改为project_dir
(2)重新安装accelerate包,降级为0.18.0版本

解决方案

方案一

修改代码将Accelerator初始化时的logging_dir修改为project_dir
修改前代码

accelerator = Accelerator(
    gradient_accumulation_steps=1,
    mixed_precision=args.mixed_precision,
    log_with=args.report_to,
    logging_dir=logging_dir,
    project_config=accelerator_project_config,
)

修改后代码

accelerator = Accelerator(
    gradient_accumulation_steps=1,
    mixed_precision=args.mixed_precision,
    log_with=args.report_to,
    #logging_dir=logging_dir,
    project_dir=logging_dir,
    project_config=accelerator_project_config,
)

方案二

重新安装accelerate包,降级为0.18.0版本

pip install accelerate==0.18.0
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐