AttributeError: 'BF16_Optimizer' object has no attribute 'cur_scale'

[xin0623]

The error occurred when I tried to fine-tune in single_modality

Traceback (most recent call last): File "run_finetuning.py", line 719, in main(opts, ds_init) File "run_finetuning.py", line 636, in main train_stats = train_one_epoch( File "/mnt2/xiayingxin/InternVideo/InternVideo2/single_modality/engines/engine_for_finetuning.py", line 92, in train_one_epoch loss_scale_value = get_loss_scale_for_deepspeed(model) File "/mnt2/xiayingxin/InternVideo/InternVideo2/single_modality/engines/engine_for_finetuning.py", line 22, in get_loss_scale_for_deepspeed return optimizer.loss_scale if hasattr(optimizer, "loss_scale") else optimizer.cur_scale AttributeError: 'BF16_Optimizer' object has no attribute 'cur_scale'

[Andy1621]

Can you provide your DeepSpeed version?

[xin0623]

deepspeed==0.10.1