CUDA out of memory in test loop

Question

For the following training program, training and validation are all ok. Once reach to Test method, I have CUDA out of memory. What should I change so that I hav