CUDA out of memory 오류

[rlawoons34]

batch size를 256으로 하니 위 에러가 발생하며, 128로 줄이니 위 오류가 발생되지 않습니다. 찾아보니 path에서 기본적으로 무료로 제공되는것을 넘어서서 그런거 같기도한데...혹시 해결방법이 있는지 궁금합니다.

[okgosu]

딥러닝 모델 층이 깊거나, 텐서의 크기가 커서 그런 듯 하네요....좀 줄이셔야할 듯 합니다