Обнаружение ошибки OMM при обучении модели ResNet с использованием графического процессора ⇐ Python

[Anonymous]

Я тренирую ResNet на NVidia GeForce GTX 1050 ti с 4 ГБ. Размер изображения составляет 512x320, используя 6 в качестве размера пакета, что не очень хорошо.< /p>
Показывает эту ошибку:

Код: Выделить всё

(1) Resource exhausted:  OOM when allocating tensor with shape[6,128,256,160] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc

И я пытался понять параметры тензора, первый (6) видимо - это количество эпох, Третий и последний (256,160) - это размерность изображение разделено на 2.
Я запутался. Возможно ли, что графический процессор не использует всю память?

Подробнее здесь: https://stackoverflow.com/questions/646 ... -using-gpu