# Run\_mlm.py cuda error memory after resuming a training

**URL:** </static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Frun-mlm-py-cuda-error-memory-after-resuming-a-training%2F5549%3E%5C
**Category:** 🤗Transformers\
**Created:** [April 16, 2021, 9:54am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Frun-mlm-py-cuda-error-memory-after-resuming-a-training%2F5549 "2021-04-16T09:54:11Z")\
**Posts on this page:** 1\
**Showing post:** 3

<div class="post-metadata">

**Author:** ![IamAdiSri](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/iamadisri/32/20434_2.png) [@IamAdiSri](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FIamAdiSri)%5C
**Post date:** [April 20, 2021, 12:09pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Frun-mlm-py-cuda-error-memory-after-resuming-a-training%2F5549%2F3 "2021-04-20T12:09:48Z")

</div>

This fix by @sgugger in a [recent commit](https://github.com/huggingface/transformers/pull/11318) resolved the issue for me. The fix is not in the official release yet, but you can install the bleeding edge version of the Transformers library from source using `pip install git+https://github.com/huggingface/transformers` which pulls it from GitHub.

---

_[View the full topic](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Frun-mlm-py-cuda-error-memory-after-resuming-a-training%2F5549)._