使用默认超参数在COCO-CN上微调，无法收敛

Open xiuxiuxius opened this issue 1 year ago • 4 comments

你好作者，我使用coco-cn_finetune_vit-b-16_rbt-base.sh在你们提供的处理好的COCO-CN上做微调，查看微调过程发现没有收敛，而MUGE和Flickr使用默认参数微调是收敛的，请问问题可能出在哪里呢？

仅修改了两个字段

GPUS_PER_NODE=1
export MASTER_ADDR=localhost

下面是三个数据集上的微调折线图

附上我在COCO-CN上微调的log文件 out_2024-01-11-09-33-24.log

Jan 24 '24 03:01 xiuxiuxius

最后解决了嘛

Sep 27 '24 01:09 Ever-Bright-V

最后解决了嘛

没有

Sep 27 '24 10:09 xiuxiuxius

您好，怎么生成这种折线图呀

Oct 26 '24 09:10 MoMo569377793

您好，怎么生成这种折线图呀

解析打印的日志，获取loss和step等信息，使用matplotlib画的

Oct 27 '24 09:10 xiuxiuxius