It is too hard to train
Radio4ctive opened this issue · comments
Hello sir,
I used your network and tried to train a model for crowd counting task for my master thesis project but in the project I faced RuntimeError: CUDA out of memory. I tried to decrease batch size but the problem still occurs. (Even though trying batch size = 1).
My GPU is RTX 3060 and I have 12 GB GPU RAM, can you give me some advice to handle this problem or share your train part?
Best regards.
Thank you.