stefanknegt / Probabilistic-Unet-Pytorch

A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch
Apache License 2.0
270 stars 67 forks source link

learning rate? #9

Closed suyanzhou626 closed 4 years ago

suyanzhou626 commented 4 years ago

Did you adopt the lr scheduler in the experiments ? or it's ok for 1e-4 all the time?

stefanknegt commented 4 years ago

If I remember correctly the results were slightly better when using a lr scheduler.