thomwolf 5e1207b8ad add attention to all bert models and add test

2019-06-14 16:28:25 +02:00

..

__init__.py

Release: 0.6.2

2019-04-25 21:37:47 +02:00

__main__.py

adding gpt2

2019-02-17 23:38:51 +01:00

convert_gpt2_checkpoint_to_pytorch.py

fix typo - logger info

2019-03-06 10:05:21 +01:00

convert_openai_checkpoint_to_pytorch.py

fix typo - logger info

2019-03-06 10:05:21 +01:00

convert_tf_checkpoint_to_pytorch.py

fix typo - logger info

2019-03-06 10:05:21 +01:00

convert_transfo_xl_checkpoint_to_pytorch.py

fix typo - logger info

2019-03-06 10:05:21 +01:00

file_utils.py

is python 2 happy now

2019-04-17 14:48:34 +02:00

modeling_gpt2.py

output attentions in GPT-2

2019-05-08 22:24:42 +02:00

modeling_openai.py

adding dropout to GPT-2 and embedding dropout to GPT

2019-05-08 10:41:35 +02:00

modeling_transfo_xl_utilities.py

fix typo - logger info

2019-03-06 10:05:21 +01:00

modeling_transfo_xl.py

load all models on cpu

2019-04-15 15:43:01 +02:00

modeling.py

add attention to all bert models and add test

2019-06-14 16:28:25 +02:00

optimization_openai.py

- updated docs for new LR API

2019-04-25 15:59:39 +02:00

optimization.py

- updated docs for new LR API

2019-04-25 15:59:39 +02:00

tokenization_gpt2.py

clean up in tokenization

2019-05-08 21:43:51 +02:00

tokenization_openai.py

clean up in tokenization

2019-05-08 21:43:51 +02:00

tokenization_transfo_xl.py

tokenization updates

2019-04-15 14:24:52 +02:00

tokenization.py

tokenization updates

2019-04-15 14:24:52 +02:00