-
Notifications
You must be signed in to change notification settings - Fork 78
训练时accelerate报错 #12
Description
我在训练的时候,accelerate报错gpu_ids找不到
Traceback (most recent call last):
File "/opt/anaconda3/envs/UVIT/bin/accelerate", line 8, in
sys.exit(main())
File "/opt/anaconda3/envs/UVIT/lib/python3.8/site-packages/accelerate/commands/accelerate_cli.py", line 43, in main
args.func(args)
File "/opt/anaconda3/envs/UVIT/lib/python3.8/site-packages/accelerate/commands/launch.py", line 747, in launch_command
defaults = load_config_from_file(args.config_file)
File "/opt/anaconda3/envs/UVIT/lib/python3.8/site-packages/accelerate/commands/config/config_args.py", line 64, in load_config_from_file
return config_class.from_yaml_file(yaml_file=config_file)
File "/opt/anaconda3/envs/UVIT/lib/python3.8/site-packages/accelerate/commands/config/config_args.py", line 117, in from_yaml_file
return cls(**config_dict)
TypeError: init() got an unexpected keyword argument 'gpu_ids'