Julien Chaumond
4c06893610
Fix nn.DataParallel compatibility in PyTorch 1.5 ( #4300 )
...
* Test case for #3936
* multigpu tests pass on pytorch 1.4.0
* Fixup
* multigpu tests pass on pytorch 1.5.0
* Update src/transformers/modeling_utils.py
* Update src/transformers/modeling_utils.py
* rename multigpu to require_multigpu
* mode doc
2020-05-18 20:34:50 -04:00
Julien Chaumond
f54dc3f4d5
[ci] Load pretrained models into the default (long-lived) cache
...
There's an inconsistency right now where:
- we load some models into CACHE_DIR
- and some models in the default cache
- and often, in both for the same models
When running the RUN_SLOW tests, this takes a lot of disk space, time, and bandwidth.
I'd rather always use the default cache
2020-04-30 22:30:15 -04:00
Julien Chaumond
f169957d0c
TF GPU CI ( #3085 )
...
* debug env
* Restrict TF GPU memory
* Fixup
* One more test
* rm debug logs
* Fixup
2020-03-02 15:45:25 -05:00
Julien Chaumond
cf8a70bf68
More AutoConfig tests
2020-01-11 03:43:57 +00:00
alberduris
81d6841b4b
GPU text generation: mMoved the encoded_prompt to correct device
2020-01-06 15:11:12 +01:00
alberduris
dd4df80f0b
Moved the encoded_prompts to correct device
2020-01-06 15:11:12 +01:00
Aymeric Augustin
067395d5c5
Move tests outside of library.
2019-12-22 13:47:17 +01:00
