GPU memory issue when using ResNet50 and ResNet152

[changliu816]

Hi Jamal,

Thanks for sharing the code. It is very impressive!

I have some gpu memory issues when using deep resnet architecture. I used a single GPU with 24GB memory with batch size = 32. When I wrote the ResNet 50 (Metamodule version) and trained on it, it caused running out of memory issues.

I think it is probably that I wrote ResNet 50 in the wrong way or other possible reasons you might help me name it. Would you mind also sharing your ResNet50 and ResNet152 in MetaModule version at resnet.py?

Best,
Chang