search

yaoppeng / U-Net_v2

216 stars 20 forks source link

Use of Optimizer #13

Closed yeerwen closed 10 months ago

yeerwen commented 10 months ago

Hi,

Great job on your work! I've noticed a few discrepancies between the paper and the implementation in your code. Specifically, the optimizer mentioned in the paper is Adam, but in your code, it appears to be SGD (as seen in line 102 of polypTrainer.py). Could you please clarify which optimizer is the correct one used in your research?

Thank you in advance for your response.

Best regards, Yiwen

yaoppeng commented 10 months ago

For polyp segmentation, please refer to the folder 'PolypSeg,' which used multi-scale training. I tried Adam, AdamW, SGD, and other optimizers. For polyp segmentation, Adam and AdamW perform a little better than SGD. For ISIC, they perform the same.

yeerwen commented 10 months ago

Thanks for the quick reply.