AttributeError: module 'torch.optim' has no attribute 'adam'

[evanweiguohua]

PTV2将优化器设置为adam会报AttributeError: module 'torch.optim' has no attribute 'adam'的错误，adamw是好的。 Lora则没有该问题。

[evanweiguohua]

配置如下：

模块配置， 默认启用lora

enable_deepspeed = True enable_ptv2 = True enable_lora = False enable_int8 = False # qlora int8 enable_int4 = False # qlora int4

[ssbuild]

pip list |grep torch

[evanweiguohua]

pytorch-lightning 2.0.4 torch 2.0.1 torchaudio 2.0.2 torchmetrics 1.0.0 torchvision 0.15.2

[ssbuild]

pip uninstall deep_training pip install -U git+https://github.com/ssbuild/deep_training.git

it is bug and fixed , do upgrade can solve.

[evanweiguohua]

Thanks a lot. I'll try it.