Loss go NAN when running classification model of Seaformer_B
shiyutang opened this issue · comments
shiyutang commented
wwqq commented
Try to train the model with 8 gpus.
shiyutang commented
Thanks a lot, I must forget to change to 8 after test.
Lun Zhongwang commented
Lun Zhongwang commented
Thanks a lot, I must forget to change to 8 after test.
Have you solved this problem?
shiyutang commented
Yes, It needs to be trained with 8 gpus.
Lun Zhongwang commented
Yes, It needs to be trained with 8 gpus.
Thanks.