Stefano Fiorucci f43bc562d3

refactor: replace torch.no_grad with torch.inference_mode (where possible) (#3601) ...

* try to replace torch.no_grad

* revert erroneous change

* revert other module breaking

* revert training/base

2022-11-23 09:26:11 +01:00

..

__init__.py

[CI refactoring] Categorize tests into folders (#2554)

2022-05-17 09:55:53 +01:00

_test_feature_extraction_end2end.py

feat: MultiModalRetriever (#2891)

2022-10-17 18:58:35 +02:00

test_distillation.py

[CI refactoring] Categorize tests into folders (#2554)

2022-05-17 09:55:53 +01:00

test_dpr.py

refactor: replace torch.no_grad with torch.inference_mode (where possible) (#3601)

2022-11-23 09:26:11 +01:00

test_feature_extraction.py

feat: MultiModalRetriever (#2891)

2022-10-17 18:58:35 +02:00

test_inference.py

Simplify language_modeling.py and tokenization.py (#2703)

2022-07-22 16:29:30 +02:00

test_language.py

Simplify language_modeling.py and tokenization.py (#2703)

2022-07-22 16:29:30 +02:00

test_prediction_head.py

Simplify language_modeling.py and tokenization.py (#2703)

2022-07-22 16:29:30 +02:00

test_processor_save_load.py

feat: MultiModalRetriever (#2891)

2022-10-17 18:58:35 +02:00

test_processor.py

feat: MultiModalRetriever (#2891)

2022-10-17 18:58:35 +02:00

test_question_answering.py

fix: strip whitespaces safely from FARMReader's answers (#3526)

2022-11-08 09:26:47 +01:00