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