segmentation_models icon indicating copy to clipboard operation
segmentation_models copied to clipboard

Weights loading and evaluation

Open luna9722 opened this issue 4 years ago • 0 comments

Hello everybody, I trained my model using weighted dice loss + categorical focal loss and saved the weights in the file weights.h5, now I have to load my weights in another notebook for prediction and post processing so i have to call the model and compile it with the same loss I used in taining, do I have to use the same class weights or use other class weights for testing ( in fact for training i calculated class weights with a sklearn function) please can someone answer my question so I can be assured I am doing the right thing ? image image

Thanks in advance.

luna9722 avatar May 14 '21 23:05 luna9722