GPU memory consumption increases while training

Hi @apaszke, I tried your solution but it doesn’t solve my problem. See here about my situation. Really appreciate your help, thanks!