How can I solve it?
Roll77-luo opened this issue · 5 comments
Roll77-luo commented
ChunhanLi commented
I met same IndexError when do prediction. If I removed apex mixed precesion training, prediction is fine.
Siu-Keung commented
Same problem while training model under mixed-precision mode.
Siu-Keung commented
RollingPlain commented
I met the same question when my torch version was 1.10.0
There is my solution
reinstall the whole environment
install torch=1.9.0
install apex
(I didn't use version 1.8. X because an inherent error may be reported in subsequent tasks