search

MrChenFeng / SSR_BMVC2022

SSR: An Efficient and Robust Framework for Learning with Unknown Label Noise (BMVC2022)
https://bmvc2022.mpi-inf.mpg.de/372/
MIT License
29 stars 7 forks source link

the gpu memory is out of, what model are you using? #1

Closed qiuxiaolong857 closed 1 year ago

qiuxiaolong857 commented 1 year ago

I'd like to ask a question,The size of the graphics device you are using to run the Clothing1M and WebVision data sets?I reproduced the results and found that the memory required for these two data sets was large

MrChenFeng commented 1 year ago

Hi, I'm experimenting with Nvidia 3090Ti. If you don't have enough GPU memory, try to reduce the batchsize.