I run the code with a V-100 GPU (32GB memory), but if the batch size is larger than 6 (or even small), it will raise this error. What caused this?
I run the code with a V-100 GPU (32GB memory), but if the batch size is larger than 6 (or even small), it will raise this error.
What caused this?