#torch-gpu-memory

avoiding full gpu memory occupation during training in pytorch