GPU out of memory

[LiuZhenshun]

Hi, I tried to use 8G 2080-ti to evaluate your 152-resnet model, but the memory is not enough. Could you tell the minimum Memory to run your test code?

[gkrisp98]

Hi, I am also facing the same problem. Also, I would like to ask you how did you manage to run the demo.py file without the error about the autograd function? I have rewrite some lines of code and I was wondering if the error for cuda appears due to the changes I made