In the process of program training, GPU memory leaks and GPU memory keeps increasing until out of memory. Can you figure this bug ?