GPU memory keeps increasing during training FORCE #4
Description
Hi, first of all I am very thankful for you to provide such an interesting dataset. However, I ran into an out of memory problem during the training FORCE phase. I monitored GPU usage while training, and it keeps increasing until overflow.