Pull to refresh
8K+
4
Максим Хотинский@assistentus

User

9
Rating
13
Subscribers
Send message

Побеждаем OOM в PyTorch: как обучать гигантские графы на обычной видеокарте

Level of difficultyEasy
Reading time6 min
Reach and readers11K

При обучении графных нейросетей и гигантских эмбеддингов на миллионы узлов стандартный torch.optim.SparseAdam в PyTorch моментально забивает RAM и VRAM, вызывая ошибку CUDA out of memory (OOM). В этой статье разбираем, как устроена легковесная Open Source библиотека Disk Sparse Adam (DSA), которая выносит состояния моментов на диск через memory mapping (mmap) и позволяет обучать огромные спарс-модели на обычной домашней видеокарте.

Читать далее

Information

Rating
838-th
Registered
Activity