【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
更多推荐
已为社区贡献8条内容
所有评论(0)