IVGSZ / Flash-VStream

This is the official implementation of "Flash-VStream: Memory-Based Real-Time Understanding for Long Video Streams"
https://invinciblewyq.github.io/vstream-page/
Apache License 2.0
129 stars 7 forks source link

loss suddenly dropped to 0 #24

Open szbcasia opened 1 week ago

szbcasia commented 1 week ago

Hello, I encountered an issue during fine-tuning. While training on llava_v1_5_mix665k_with_video_chatgpt, the loss suddenly dropped to 0 at a certain step and has remained at 0. Could you help me identify the possible cause of this issue? {'loss': 1.9156, 'learning_rate': 8.695652173913044e-07, 'epoch': 0.0}
{'loss': 1.8055, 'learning_rate': 1.7391304347826088e-06, 'epoch': 0.0}
{'loss': 0.0, 'learning_rate': 2.6086956521739132e-06, 'epoch': 0.0}
{'loss': 0.0, 'learning_rate': 3.4782608695652175e-06, 'epoch': 0.01}
{'loss': 0.0, 'learning_rate': 4.347826086956522e-06, 'epoch': 0.01}
{'loss': 0.0, 'learning_rate': 5.2173913043478265e-06, 'epoch': 0.01}
{'loss': 0.0, 'learning_rate': 6.086956521739132e-06, 'epoch': 0.01}

zhang9302002 commented 1 week ago

Hello, thanks for your attention!

We didn't meet this problem ever. Maybe you could check your python environment and output some middle variables during training for debugging.

Best.