Superlee506 / Mask_RCNN_Humanpose

Mask R-CNN for Human Pose Estimation on Keras and TensorFlow.
Other
189 stars 36 forks source link

I found that the val_loss can not converge while the train_loss can converge #17

Closed whytin closed 6 years ago

whytin commented 6 years ago

Thank you first for your nice work! I use this model to train the new datasets (Penn Action), and I found that the val_loss can not converge while the train_loss can converge. I did not modify the config and the process. I want to figure out what causes this problem. Thanks! The train_loss convergence: screenshot from 2018-04-23 20-24-30 The val_loss convergence: screenshot from 2018-04-23 20-27-52

Superlee506 commented 6 years ago

@whytin In your example, over-fitting happens. You can try the following methods: 1) Add more training data or data augment; 2) increase the weight decay; 3) simplify the model

whytin commented 6 years ago

I will try, Thanks!