hpcaitech / ColossalAI

Making large AI models cheaper, faster and more accessible
https://www.colossalai.org
Apache License 2.0
38.79k stars 4.34k forks source link

[BUG]: Colossalai-OpenMoE-8b : loss value is very large and cannot converge #5212

Open hangchen426926 opened 10 months ago

hangchen426926 commented 10 months ago

🐛 Describe the bug

I am currently running the Colossalai/examples/language/openmoe project with the following experimental setup:

datasets: load_dataset("yizhongw/self_instruct/data/finetuning/self_instruct_221203", "super_natural_instructions"), model: openmoe-8b A100_num_GPU:8 Epoch:3 BatchSize:4 LR:0.00001 zero_stage:1 precision:bf16 Boost plugin:ep_zero extra_dp_size:2 max_length:2048

### Issue : loss value is very large and cannot converge I've encountered challenges during the training process where the convergence seems unachievable. The training loss value persists at an exceptionally high level (exceeding 2.3E+10) even after running 3 epochs. The logged information provided below showcases loss values:

loss

Furthermore, in the default setup of the openmoe project, the lr is 0.00001. Considering this, I suspect the issue might stem from an overfitting problem. Consequently, I attempted to adjustment the lr parameter(from 0.00001 to 0.0000001) ,but still encountered the same problem. tempsnip tempsnip2

Environment

torch 1.13.1 python 3.8.17 cuda:11.7.17

Orion-Zheng commented 10 months ago

Thank you for your valuable feedback! 😃We are working on this bug and will get back to you later.

noob-ctrl commented 10 months ago

I also encountered this problem

noob-ctrl commented 10 months ago

@Orion-Zheng Has this bug been solved?