search

yaoppeng / U-Net_v2

211 stars 18 forks source link

Use of Optimizer #13

Closed yeerwen closed 9 months ago

yeerwen commented 9 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 9 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 9 months ago

Thanks for the quick reply.