Когда я тренирую модель с 512*512 изображениями с batch_size=4, она работает, но когда я тренирую ее с 1024*1024 изображениями с batch_size=1, возникает ошибка cuda Недостаточно памяти.
Я запутался, общее количество пикселей равно.Почему одно 1024*1024 изображение использует больше памяти, чем четыре 512*512 изображения?