tests/test_modeling_transfo_xl.py · 6b410bedfc93269083eef7a712ef686e22e93b23 · chenpangpang / transformers

Shift labels internally within TransfoXLLMHeadModel when called with labels (#3716) · 352d5472

Teven authored Apr 13, 2020



* Shifting labels inside TransfoXLLMHead

* Changed doc to reflect change

* Updated pytorch test

* removed IDE whitespace changes

* black reformat
Co-authored-by: TevenLeScao <teven.lescao@gmail.com>

352d5472

test_modeling_transfo_xl.py 17 KB

Replace test_modeling_transfo_xl.py