View of Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet

Return to Article Details Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet Download Download PDF