Thx, I got you. But I have another question.
In this project, my validation set is only one fifth of training set, so I think it’s weird that validation phase arises CUDA out of memory but training phase won’t instead.