A implementation of ResNet-18 uses lot of GPU memory

Thanks! All give that a shot.