Pytorch-UNet icon indicating copy to clipboard operation
Pytorch-UNet copied to clipboard

Dice score is not change

Open JonyJiang123 opened this issue 3 years ago • 4 comments

Recently, I got the problem when trained model with own data.

image

Dice score is unchanged and close to 0, but the same data is in another unet version(keras), test result is good.

INFO: Starting training: Epochs: 50 Batch size: 16 Learning rate: 0.0001 Training size: 232 Validation size: 25 Checkpoints: True Device: cuda Images scaling: 1.0 Mixed Precision: False

JonyJiang123 avatar Oct 15 '21 07:10 JonyJiang123

You can try changing the optimizer

Huster-Hq avatar Mar 10 '22 14:03 Huster-Hq

have you solved it yet? help me

nqthai309 avatar Jun 05 '22 16:06 nqthai309

@biel-jony, try to reduce learning rate and check result.

VladDok avatar Jan 04 '23 23:01 VladDok

Hello, did you manage to solve it?

leosmaia21 avatar Apr 22 '23 23:04 leosmaia21