RU

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

При обучении графных нейросетей и гигантских эмбеддингов на миллионы узлов стандартный torch.optim.SparseAdam в PyTorch моментально забивает RAM и V…

PyTorchMachine LearningOpen SourcePythonGNNOOMSparseAdamОптимизация памятиDisk Sparse AdamГлубокое обучение
Habr
RU

Сжать четыре токена в один вектор: запускаем автоэнкодер CALM на доменных данных (и на одном CPU)

Взяли автоэнкодер из свежей работы  CALM (Continuous Autoregressive Language Models) , который учится упаковывать чанк из K=4 токенов в один не…

CALMАвтоэнкодерМашинное обучениеГлубокое обучениеNLPLLMCPU