Optimizer for unet model on Pascal Voc segmentation

warmspringwinds / pytorch-segmentation-detection

Image Segmentation and Object Detection in Pytorch

MIT License

746 stars 170 forks source link

Optimizer for unet model on Pascal Voc segmentation #24

Open saiphanish7 opened 4 years ago

saiphanish7 commented 4 years ago

Hello, Can I know the optimizer and its specifications to use on unet model on Pascal Voc segmentation using FOCAL loss ? Should I have to use any learning rate schedulers? Also, is it better to take mean focal loss or sum focal loss ? I am training the model from scratch.