riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity
e03966e404
transformers/tests/rag
History
Yih-Dar 32adbb26d6
Fix PyTorch RAG tests GPU OOM (#16881) 
* add torch.cuda.empty_cache in some PT RAG tests

* torch.cuda.empty_cache in tearDownModule()

* tearDown()

* add gc.collect()

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-04-25 17:33:56 +02:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_modeling_rag.py Fix PyTorch RAG tests GPU OOM (#16881) 2022-04-25 17:33:56 +02:00
test_modeling_tf_rag.py Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
test_retrieval_rag.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_tokenization_rag.py Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00