Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

4卡3090运行 #6

Open
shayeyty opened this issue Sep 25, 2024 · 13 comments
Open

4卡3090运行 #6

shayeyty opened this issue Sep 25, 2024 · 13 comments

Comments

@shayeyty
Copy link

作者您好,我在4卡3090环境上配置SDCL并在SYSU数据集上复现多次都在(indoor:67%,All:61%)达不到您的效果(indoor: 71%,All: 64%),我想请教一下您这可能是什么原因导致的,其中训练时epoch1为(r1: 19.7% best: 15.8%),epoch49为(r1: 66.9% best: 61.2%)。

@yangbincv
Copy link
Owner

感谢对我们工作的关注,请问您的batch size是多少,batchsize会有一些影响。此外,代码环境也会有影响,之前交流的有几位同学都复现出来了论文的结果,方便的话,留一个邮箱,我给您发一个我们训练的log,供您参考一下,做进一步的讨论和支持

@shayeyty
Copy link
Author

作者您好,我复现的代码只改过数据集存放的路径,batchSize是遵循train_cc_vit_sysu.sh中的256大小,代码运行环境可能和您有点差异。我的邮箱是:[email protected],非常感谢您的回复。

@unicofu
Copy link

unicofu commented Oct 8, 2024

@shayeyty 你好,我也遇到了这个问题,请问你解决了吗

@yangbincv
Copy link
Owner

@unicofu 你好,我上传了两个数据集的训练和测试的log,此外也上传了代码的环境版本要求,供您参考。有任何复现问题可以随时联系[email protected],希望能帮助你解决问题

@LqhThird
Copy link

@unicofu @shayeyty 你好,我用的也是4张3090,还是同样的问题,请问下现在找到办法解决了吗

@yangbincv
Copy link
Owner

@LqhThird 可以尝试调一调环境版本和训练的batchsize之类的参数,也欢迎各位大佬分享解决方案 @shayeyty @unicofu

@shayeyty
Copy link
Author

@yangbincv 作者您好,请问能提供所提出模型的model_best.pth吗?我们在复现过程出现问题,但是我们想在我们的论文中和您的方法进行对比实验。

@yangbincv
Copy link
Owner

好的,我发给您邮箱 @shayeyty

@yangbincv
Copy link
Owner

@unicofu @LqhThird @shayeyty 另外有任何问题都可以邮件给我,或者加vx,我会积极支持各位复现该工作

@shayeyty
Copy link
Author

感谢作者 @yangbincv

@yangbincv
Copy link
Owner

@unicofu @LqhThird @shayeyty 大家好,我找了一台4张v100卡的服务器重新调整了参数(num-instances调整为4,trainer.hm=1),同样达到了和论文中差不多的效果,文件已经上传,见train_cc_vit_sysu_gpux4.sh和cluster_contrast_vit_sysu_gpux4.py,有任何问题可以联系我。

@shayeyty
Copy link
Author

@yangbincv 通过您指导的调整,我已经能在4卡服务器上达到与论文中差不多的效果了,我单次的复现结果是SYSU-MM01中Indoor Search下Rank-1:70.66%,mAP:75.90%,mINP:72.46%,All Search下Rank-1:65.64%,mAP:63.98,mINP:51.39%。感谢您一直以一来的关注的和指导。

@yangbincv
Copy link
Owner

@shayeyty 🎉🎉🎉🎉🎉棒,有问题随时交流

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants