thomwolf
|
d821358884
|
update readme
|
2018-12-14 15:15:17 +01:00 |
|
thomwolf
|
37378898a2
|
adding DockerFile
|
2018-12-14 15:02:32 +01:00 |
|
thomwolf
|
4a4b0e5783
|
remove logging. basicConfig from library code
|
2018-12-14 14:46:25 +01:00 |
|
thomwolf
|
ae88eb88a4
|
set encoding to 'utf-8' in calls to open
|
2018-12-14 13:48:58 +01:00 |
|
thomwolf
|
e1eab59aac
|
no fp16 on evaluation
|
2018-12-13 14:54:02 +01:00 |
|
thomwolf
|
087798b7fa
|
fix reloading model for evaluation in examples
|
2018-12-13 14:48:12 +01:00 |
|
thomwolf
|
0f544625f4
|
fix swag example for work with apex
|
2018-12-13 13:35:59 +01:00 |
|
thomwolf
|
0cf88ff084
|
make examples work without apex
|
2018-12-13 13:28:00 +01:00 |
|
thomwolf
|
52c53f39d0
|
clean up apex integration
|
2018-12-13 13:02:17 +01:00 |
|
thomwolf
|
4946c2c500
|
run_swag example in readme
|
2018-12-13 13:02:07 +01:00 |
|
thomwolf
|
d23eed85bb
|
model loading apex modification
|
2018-12-13 12:53:17 +01:00 |
|
thomwolf
|
1cbb32a542
|
include version number + comment in setup.py
|
2018-12-13 12:50:44 +01:00 |
|
thomwolf
|
ce52177638
|
added version in __init__.py
|
2018-12-13 12:50:44 +01:00 |
|
thomwolf
|
d3fcec1a3e
|
add saving and loading model in examples
|
2018-12-13 12:50:44 +01:00 |
|
thomwolf
|
93f335ef86
|
add pretrained loading from state_dict
|
2018-12-13 12:48:13 +01:00 |
|
thomwolf
|
b3caec5a56
|
adding save checkpoint and loading in examples
|
2018-12-13 12:48:13 +01:00 |
|
thomwolf
|
85fff78c2d
|
compatibility PT 1.0 and 0.4.1
|
2018-12-13 12:48:13 +01:00 |
|
thomwolf
|
13bf0d4659
|
fixing Adam weights skip in TF convert script
|
2018-12-13 12:48:13 +01:00 |
|
thomwolf
|
1df6f26214
|
Merge branch 'fourth-release' of https://github.com/huggingface/pytorch-pretrained-BERT into fourth-release
|
2018-12-11 12:20:31 +01:00 |
|
thomwolf
|
770f805ae5
|
include version number + comment in setup.py
|
2018-12-11 12:20:22 +01:00 |
|
thomwolf
|
ed3b62cd3b
|
added version in __init__.py
|
2018-12-11 12:12:08 +01:00 |
|
thomwolf
|
b13abfa9fe
|
add saving and loading model in examples
|
2018-12-11 11:58:07 +01:00 |
|
thomwolf
|
270fa2f20b
|
add pretrained loading from state_dict
|
2018-12-11 11:50:38 +01:00 |
|
thomwolf
|
174cdbccde
|
adding save checkpoint and loading in examples
|
2018-12-09 17:04:23 -05:00 |
|
thomwolf
|
1db916b5be
|
compatibility PT 1.0 and 0.4.1
|
2018-12-09 16:57:51 -05:00 |
|
thomwolf
|
68f77303b2
|
fixing Adam weights skip in TF convert script
|
2018-12-09 16:17:11 -05:00 |
|
thomwolf
|
f9f3bdd60b
|
update readme
|
2018-11-30 23:05:18 +01:00 |
|
thomwolf
|
52ff0590ff
|
tup => tpu
|
2018-11-30 23:01:10 +01:00 |
|
thomwolf
|
511bce58bd
|
update new token classification model
|
2018-11-30 22:56:02 +01:00 |
|
thomwolf
|
258eb50086
|
bump up version
|
2018-11-30 22:55:33 +01:00 |
|
thomwolf
|
d787c6be8c
|
improve docstrings and fix new token classification model
|
2018-11-30 22:55:26 +01:00 |
|
thomwolf
|
ed302a73f4
|
add new token classification model
|
2018-11-30 22:55:03 +01:00 |
|
thomwolf
|
89d47230d7
|
clean up classification model output
|
2018-11-30 22:54:53 +01:00 |
|
thomwolf
|
7f7c41b0c1
|
tests for all model classes with and without labels
|
2018-11-30 22:54:33 +01:00 |
|
thomwolf
|
257a35134a
|
fix pickle dump in run_squad example
|
2018-11-30 14:23:09 +01:00 |
|
thomwolf
|
c588453a0f
|
fix run_squad
|
2018-11-30 14:22:40 +01:00 |
|
thomwolf
|
d6f06c03f4
|
fixed loading pre-trained tokenizer from directory
|
2018-11-30 14:09:06 +01:00 |
|
thomwolf
|
532a81d3d6
|
fixed doc_strings
|
2018-11-30 13:57:01 +01:00 |
|
thomwolf
|
296f006132
|
added BertForTokenClassification model
|
2018-11-30 13:56:53 +01:00 |
|
thomwolf
|
298107fed7
|
Added new bert models
|
2018-11-30 13:56:02 +01:00 |
|
thomwolf
|
0541442558
|
add do_lower_case in examples
|
2018-11-30 13:47:33 +01:00 |
|
thomwolf
|
32167cdf4b
|
remove convert_to_unicode and printable_text from examples
|
2018-11-26 23:33:22 +01:00 |
|
thomwolf
|
ce37b8e481
|
bump version in setup.py
|
2018-11-26 10:45:48 +01:00 |
|
thomwolf
|
05053d163c
|
update cache_dir in readme and examples
|
2018-11-26 10:45:13 +01:00 |
|
thomwolf
|
63ae5d2134
|
added cache_dir option in from_pretrained
|
2018-11-26 10:21:56 +01:00 |
|
thomwolf
|
029bdc0d50
|
fixing readme examples
|
2018-11-26 09:56:41 +01:00 |
|
thomwolf
|
ebaacba38b
|
fixing typo in docstring
|
2018-11-26 09:55:15 +01:00 |
|
thomwolf
|
870d71636e
|
fixing target size in crossentropy losses
|
2018-11-26 09:51:34 +01:00 |
|
thomwolf
|
982339d829
|
fixing unicode error
|
2018-11-23 12:22:12 +01:00 |
|
thomwolf
|
6b2136a8a9
|
fixing weights decay in run_squad example
|
2018-11-20 10:12:44 +01:00 |
|
thomwolf
|
eed255a58d
|
fixing CLI typo in readme
|
2018-11-20 10:02:57 +01:00 |
|
thomwolf
|
2f21497d3e
|
fixing param.grad is None in fp16 examples
|
2018-11-20 10:01:21 +01:00 |
|
thomwolf
|
27ee0fff3c
|
add no_cuda args in extract_features
|
2018-11-17 23:04:44 +01:00 |
|
thomwolf
|
aa50fd196f
|
remove unused arguments in example scripts
|
2018-11-17 23:01:05 +01:00 |
|
thomwolf
|
47a7d4ec14
|
update examples from master
|
2018-11-17 12:21:35 +01:00 |
|
thomwolf
|
c8cba67742
|
clean up readme and examples
|
2018-11-17 12:19:16 +01:00 |
|
thomwolf
|
757750d6f6
|
fix tests
|
2018-11-17 11:58:14 +01:00 |
|
thomwolf
|
a99b971738
|
bump up version minor
|
2018-11-17 10:43:39 +01:00 |
|
thomwolf
|
4e46affc34
|
updating examples
|
2018-11-17 10:30:54 +01:00 |
|
thomwolf
|
d0673c7dbd
|
fix links
|
2018-11-17 08:59:29 +01:00 |
|
thomwolf
|
68b937aa40
|
sub section overviews
|
2018-11-17 08:55:56 +01:00 |
|
thomwolf
|
c54d8b1847
|
fixing links in readme
|
2018-11-17 08:46:17 +01:00 |
|
thomwolf
|
f920eff8c3
|
update readme
|
2018-11-17 08:42:45 +01:00 |
|
thomwolf
|
886cb49792
|
updating readme and notebooks
|
2018-11-16 14:31:15 +01:00 |
|
thomwolf
|
fd647e8c87
|
comparison masked LM ok
|
2018-11-16 11:04:31 +01:00 |
|
thomwolf
|
02173a1a0a
|
fixing error in isnan test for optimizer_on_cpu & fp16
|
2018-11-15 21:49:12 +01:00 |
|
thomwolf
|
cba85a67b9
|
fix nan in optimizer_on_cpu
|
2018-11-15 21:47:41 +01:00 |
|
thomwolf
|
1de35b624b
|
preparing for first release
|
2018-11-15 20:56:10 +01:00 |
|
thomwolf
|
d940eeda54
|
typo
|
2018-11-12 15:26:46 +01:00 |
|
thomwolf
|
1cf0a16c67
|
cleaning up readme
|
2018-11-12 15:24:47 +01:00 |
|
thomwolf
|
66b0090877
|
add fp16 training
|
2018-11-12 15:15:02 +01:00 |
|
thomwolf
|
a81a1ef8e9
|
fixing learning rate schedule when using gradient_accumulation_steps
|
2018-11-10 16:11:14 +01:00 |
|
thomwolf
|
ea85cca8ab
|
adding optimize_on_cpu explanation in readme
|
2018-11-09 11:42:37 +01:00 |
|
thomwolf
|
5f04aa00ed
|
option to perform optimization and keep the optimizer averages on CPU
|
2018-11-09 11:28:14 +01:00 |
|
thomwolf
|
9e95cd8cd6
|
clean up optimizer from unused functions
|
2018-11-09 11:23:55 +01:00 |
|
thomwolf
|
34a1a01091
|
update code comment
|
2018-11-09 09:31:20 +01:00 |
|
thomwolf
|
34bdc8b54f
|
remove duplicate accumulate gradient step arguments
|
2018-11-09 09:19:45 +01:00 |
|
thomwolf
|
2c5d993ba4
|
update readme - fix SQuAD model on multi-GPU
|
2018-11-08 21:22:22 +01:00 |
|
thomwolf
|
48d4a5317c
|
typo fix in output tuple
|
2018-11-07 23:51:12 +01:00 |
|
thomwolf
|
dbc318a4c6
|
cleaning up - speeding up a bit multi-gpu
|
2018-11-07 22:22:55 +01:00 |
|
thomwolf
|
6bb7510a50
|
fixing pre-processing bug - averaging loss for gradient accumulation - no_grad on evaluation
|
2018-11-07 22:12:41 +01:00 |
|
thomwolf
|
a1126237a9
|
clean up logits extraction logic
|
2018-11-06 17:31:15 +01:00 |
|
thomwolf
|
2a97fe220b
|
fixing weights initialization in the model and out of span clamping
|
2018-11-06 17:26:33 +01:00 |
|
thomwolf
|
907d3569c1
|
cleaning up SQuAD notebook - more explanation - fixing error
|
2018-11-06 11:13:43 +01:00 |
|
thomwolf
|
88e793f31a
|
fix typos
|
2018-11-05 16:14:19 +01:00 |
|
thomwolf
|
3914eed505
|
update readme
|
2018-11-05 16:09:27 +01:00 |
|
thomwolf
|
bab5d13077
|
update optimizer documentation
|
2018-11-05 16:09:21 +01:00 |
|
thomwolf
|
7394eb47a5
|
update readme
|
2018-11-05 15:35:44 +01:00 |
|
thomwolf
|
e6646751ac
|
update notebooks
|
2018-11-05 15:02:50 +01:00 |
|
thomwolf
|
b705c9eff5
|
remove small script, moved notebooks to notebook folder
|
2018-11-05 14:55:08 +01:00 |
|
thomwolf
|
3a301d443b
|
update gitignore
|
2018-11-05 14:53:43 +01:00 |
|
thomwolf
|
711d3f9f2b
|
remove tensorflow_code
|
2018-11-05 14:53:03 +01:00 |
|
thomwolf
|
7875b1a8e0
|
notebook update
|
2018-11-05 14:50:44 +01:00 |
|
thomwolf
|
c3527cfbc4
|
ignore SQuAD targets outside of seq_length
|
2018-11-05 14:18:48 +01:00 |
|
thomwolf
|
1b99cdf71b
|
script that use a small portion of squad only
|
2018-11-05 13:54:54 +01:00 |
|
thomwolf
|
2f4765d3ed
|
fix multi-gpu squad loss
|
2018-11-05 13:46:14 +01:00 |
|
thomwolf
|
955cee33a5
|
updating SQuAD comparison
|
2018-11-05 13:21:53 +01:00 |
|
thomwolf
|
5622d8320f
|
allowing to load small number of examples
|
2018-11-05 13:21:24 +01:00 |
|
thomwolf
|
a725db4f6c
|
fixing BertForQuestionAnswering loss computation
|
2018-11-05 13:21:11 +01:00 |
|
thomwolf
|
bb5ce67a14
|
adding back tf code + adding models comparison on SQuAD
|
2018-11-05 12:11:32 +01:00 |
|