reduce memory usage for backwards layer
try to train a variant of VGG model this year