Код Junior PyTorch на практике обновлено
Параметры и память MLP
Условие
MLP задан списком размеров слоёв, например [784, 256, 10]: полносвязные слои со смещением между соседними размерами. Напишите mlp_stats(sizes, bytes_per_param): верните [params, memory_mb], где memory_mb это память на веса в мегабайтах (1 МБ = 2²⁰ байт), округлённая до 4 знаков.
Примеры
mlp_stats([784, 256, 10], 4) // [203530, 0.7764]
mlp_stats([10, 1], 4) // [11, 0]
mlp_stats([4096, 4096, 4096], 2) // [33562624, 64.0156]
Решите в тренажёре
Редактор с автотестами прямо в браузере. Можно решить спокойно, вслух или под таймером, как на собесе.