fix Lightning failing Bagua test for manual optimization
wangraying opened this issue · comments
see this：Lightning-AI/pytorch-lightning#16225
After an investigation of this problem, I found that this failure is caused by scikit-optimize library used by Bagua. In detail, scikit-optimize library currently uses an old builtin type of numpy (i.e., numpy.int), which is deprecated by numpy>=1.24.0.
To avoid this problem, users can choose to use numpy<1.24.0.
In the meantime, scikit-optimize is trying to fix this problem (PR: scikit-optimize/scikit-optimize#1123). After that, users can use the updated scikit-optimize with numpy>=1.24.0 in Bagua.