Я работаю с Keras 2.0.0, и я бы хотел обучить глубокую модель с огромным количеством параметров на графическом процессоре.
Используя слишком большие изображения, у меня заканчивается память (OOM).
Используя слишком низкие изображения, точность модели будет хуже, чем возможно.
Поэтому я хотел бы найти максимально возможный размер входных изображений, подходящих для моего графического процессора.
Есть ли какая-либо функциональность, вычисляющая память (например, сравнимую с model.summary()
) с учетом модели и входных данных?
Я ценю вашу помощь.