allenai / allenai/olmocr

OOM occurs during GRPO training on 4× H100 (80GB)

Ouverte
#413 0 commentaires 0 réactions 0 personnes assignées Voir sur GitHub
Langage dominant
Python
Étoiles
19.5k
Forks
1.6k
Métriques de merge des PR
Aucune PR mergée en 30 j

Description

I am encountering out-of-memory (OOM) errors during training on H100 (80GB) GPUs.

My understanding is that the model is not explicitly sharded, so it should be able to run on 4 GPUs. However, OOM still occurs during GRPO training. Is there something I might be missing in terms of configuration or setup?

Could you clarify the minimum GPU requirements for GRPO training (e.g., number of GPUs and memory per GPU)? Any guidance or suggestions would be greatly appreciated.

Guide de contribution

Ouvrir le guide de contribution

Évaluation

Cette issue n'a pas encore été évaluée.

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.