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