RU

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

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

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