Gpt-oss training on A100 - OOM error

DhineshR
Thu, 20 Nov 2025 18:31:11 +0000

Do you want to train in fp32 precision? If not required, try bf16 precision. I see convert_to_fp32 in the stack trace and hence this suggestion.

Also, make sure the training data shapes are as expected.