juhongm999 / hsnet

Official PyTorch Implementation of Hypercorrelation Squeeze for Few-Shot Segmentation, ICCV 2021
231 stars 43 forks source link

Validation loss #26

Closed viethoang303 closed 2 years ago

viethoang303 commented 2 years ago

Hello authors and everyone.

I trained HSnet on my own dataset about echocardigraphy images, but validation loss fluctuates a lot and model converge early. I trained model with learning rate 1e-3 and I looked Fig 9 in paper about this model, validation loss also take 10-30 epochs to achieve higher validation IoU

Validation-loss-hsnet_1shot_on_3C

So, what problem I got. Please help me.

Thank

juhongm999 commented 2 years ago

Perhaps the size of your dataset is too small, making the model converge quickly. How many images are there in your dataset?