A implementation of ResNet-18 uses lot of GPU memory

Specific Domains Machine Learning
,
3

Thanks! All give that a shot.