search

microsoft / ptgnn

A PyTorch Graph Neural Network Library
MIT License
375 stars 40 forks source link

Trainer option to catch occasional CUDA OOMs #18

Closed mallamanis closed 3 years ago

mallamanis commented 3 years ago

This should allow robust training on variable-sized data where OOMs happen rarely but interrupt training.