search

yangbincv / SDCL

MIT License
10 stars 2 forks source link

关于复现内存问题 #4

Open MYL1227 opened 1 month ago

MYL1227 commented 1 month ago

作者您好，想请教一下这篇文献需要多大显存的服务器，目前用24g显存的服务器batch调整到16还是爆显存

yangbincv commented 1 month ago

可能你调错了地方，batch-size通过代码中的cluster_contrast_vit_sysu.py中的 ir_batch，rgb_batch参数调整