A implementation of ResNet-18 uses lot of GPU memory

@OTapio could you open an issue on Flux?