$ 0 0 I successfully trained the network but got this error during validation:RuntimeError: CUDA error: out of memory