[CUDA out of memory] ResNet18 & Cifar100
lhchau opened this issue · comments
Hoang-Chau Luong commented
Hi logancyang, I have successfully run your code using ResNet18 on Cifar100 with bs=16, epochs=200. However, during the 4th epoch, the GPU VRAM overflowed. Upon reviewing the code, I found that this line is causing the GPU RAM to accumulate at each step. Do you have a solution for this problem?
Logan Yang commented
hi @lhchau, thanks for checking this out. This project is not actively maintained at the moment, but feel free to make your own fork if you have time to implement a solution.