python

HuggingFace Trainer OOM During Training

OutOfMemoryError.*during.*Trainer\.train\(\)

Fixes

1.Set per_device_train_batch_size=1 with gradient_accumulation_steps
2.Enable fp16=True or bf16=True in TrainingArguments
3.Use gradient_checkpointing=True in TrainingArguments

huggingfacetraineroom

Related Errors

Asyncio event loop already running

RuntimeError: This event loop is already running

•Use nest_asyncio.apply() to allow nested event loops
•Use asyncio.run_coroutine_threadsafe() instead of asyncio.run()

Coroutine never awaited

RuntimeWarning: coroutine '.*' was never awaited

•Add 'await' before the coroutine call
•Use asyncio.create_task() to schedule the coroutine

Asyncio task was cancelled

asyncio\.CancelledError

•Handle CancelledError in try/except within the task
•Use asyncio.shield() to protect critical sections from cancellation