关于实验所需的gpu

ls1rius / WSI_FiVE

Generalizable Whole Slide Image Classification with Fine-Grained Visual-Semantic Interaction

21 stars 1 forks source link

Open du-67 opened 4 hours ago

du-67 commented 4 hours ago

您好，又来麻烦您了。请问few-shot实验需要多少内存的gpu呢？（非端到端的实验）我目前只有一块40g的A100，能进行实验吗？运行时显示内存不足，请问我应该更改哪个文件的参数使实验顺利运行呢？期待您的回复，非常感谢!

ls1rius commented 4 hours ago

可以运行

可以通过减小NUM_FRAMES最大采样数，BATCH_SIZE，ACCUMULATION_STEPS梯度累积次数同时增大些NUM_FRAMES_PERCENT采样率来提高性能稳定性。

du-67 commented 4 hours ago

好的好的，非常感谢您的快速回复！