[ Colab 오류] Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info. This isn't available when running in Eager mode.

문제 상황

 

 

모델 학습시키는 과정에서 아래의 오류가 발생하였다.

 

If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info. This isn't available when running in Eager mode.

 

---

해결 방법

1. 아래의 소스 코드를 입력한다.

run_opts = tf.RunOptions(report_tensor_allocations_upon_oom = True)

하지만 tensorflow 버전 문제로 탈락.

 

2. batch size 조정

 

위의 오류는 메모리 부족 문제이다. batch size를 적절하게 줄여주면 해결된다.

 

- 참고링크 : https://www.py4u.net/discuss/191132