【BUG】TypeError: Accelerator.__init__() got an unexpected keyword argument ‘logging_dir‘

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
  • 10
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值