#gpu-memory-full

avoiding full gpu memory occupation during training in pytorch