2021 이전 자료들/AI

딥러닝 실험 일지

.log 2020. 2. 4. 16:34
728x90

RuntimeError: CUDA out of memory. Tried to allocate 600.00 MiB (GPU 0; 7.80 GiB total capacity; 5.92 GiB already allocated; 255.12 MiB free; 5.21 MiB cached)

 

이 에러가 났다.

네트워크 별로 넣는 input의 형태가 다른데, 1채널이었던 것을 3채널로 증가시켰더니 이렇게 나왔다.

에러를 해결하기 위해서 네트워크의 처리 사이즈를 조금 줄였다.

 

음 그리고 트레이닝을 돌렸는데, 결과는 가봐야 알 수 있을 것이다.