thomwolf
|
c41f2bad69
|
WIP XLM + refactoring
|
2019-07-03 22:54:39 +02:00 |
|
thomwolf
|
288be7b7ea
|
xlm
|
2019-07-02 23:42:31 +02:00 |
|
thomwolf
|
708877958a
|
updating tests and models, adding weights initialization test
|
2019-07-02 16:35:29 +02:00 |
|
thomwolf
|
99ae5ab883
|
update config tests and circle-ci
|
2019-07-02 12:40:39 +02:00 |
|
thomwolf
|
1484d67de9
|
[LARGE] updating all tests and API
|
2019-07-02 12:13:17 +02:00 |
|
thomwolf
|
4f8b5f687c
|
add fix for serialization of tokenizer
|
2019-06-29 23:35:21 +02:00 |
|
thomwolf
|
d9184620f9
|
fix tests and new API
|
2019-06-29 23:10:40 +02:00 |
|
thomwolf
|
213981d8cb
|
updating bert API
|
2019-06-28 16:45:24 +02:00 |
|
thomwolf
|
2b56e98892
|
standardizing API across models - XLNetForSeqClass working
|
2019-06-28 16:35:09 +02:00 |
|
thomwolf
|
3a00674cbf
|
fix imports
|
2019-06-27 17:18:46 +02:00 |
|
thomwolf
|
d939d6fd02
|
fix hidden-state extraction
|
2019-06-27 09:39:44 +02:00 |
|
thomwolf
|
0c2ff34815
|
extracting double hidden-state from xlnet
|
2019-06-27 09:27:50 +02:00 |
|
thomwolf
|
3deea56c07
|
fixing loading fucntion
|
2019-06-26 13:41:12 +02:00 |
|
thomwolf
|
f56b8033f0
|
more versatile loading
|
2019-06-26 13:13:15 +02:00 |
|
thomwolf
|
4d47f4985d
|
slight refactoring, add abstract class for model loading
|
2019-06-26 12:52:44 +02:00 |
|
thomwolf
|
59cefd4f98
|
fix #726 - get_lr in examples
|
2019-06-26 11:28:27 +02:00 |
|
thomwolf
|
ddc2cc61a6
|
fix python2 tests
|
2019-06-26 11:17:42 +02:00 |
|
thomwolf
|
7e3070ae4f
|
add from_pretrained method to all configuration classes
|
2019-06-26 11:12:00 +02:00 |
|
thomwolf
|
93e9971c54
|
fix tests
|
2019-06-26 10:02:45 +02:00 |
|
thomwolf
|
092dacfd62
|
changing is_regression to unified API
|
2019-06-26 09:54:05 +02:00 |
|
thomwolf
|
e55d4c4ede
|
various updates to conversion, models and examples
|
2019-06-26 00:57:53 +02:00 |
|
thomwolf
|
603c513b35
|
update main conversion script and readme
|
2019-06-25 10:45:07 +02:00 |
|
thomwolf
|
7de1740490
|
add ability to restore fine-tuned TF mdoel
|
2019-06-25 10:27:58 +02:00 |
|
thomwolf
|
7334bf6c21
|
pad on left for xlnet
|
2019-06-24 15:05:11 +02:00 |
|
thomwolf
|
c888663f18
|
overwrite output directories if needed
|
2019-06-24 14:38:24 +02:00 |
|
thomwolf
|
62d78aa37e
|
updating GLUE utils for compatibility with XLNet
|
2019-06-24 14:36:11 +02:00 |
|
thomwolf
|
24ed0b9346
|
updating run_xlnet_classifier
|
2019-06-24 12:00:09 +02:00 |
|
thomwolf
|
f6081f2255
|
add xlnetforsequence classif and run_classifier example for xlnet
|
2019-06-24 10:01:07 +02:00 |
|
thomwolf
|
c946bb51a6
|
fix xlnet tokenizer and python2
|
2019-06-22 22:28:49 +02:00 |
|
thomwolf
|
181075635d
|
updating model loading and adding special tokens ids
|
2019-06-21 23:23:37 +02:00 |
|
thomwolf
|
ebd2cb8d74
|
update from_pretrained to load XLNetModel as well
|
2019-06-21 21:08:44 +02:00 |
|
thomwolf
|
483cbc36a9
|
test deviation with tf model: max ~1e-3 should be ok
|
2019-06-21 16:38:01 +02:00 |
|
thomwolf
|
24d8068982
|
weights loading script ok
|
2019-06-21 12:33:44 +02:00 |
|
thomwolf
|
32da75486b
|
add tokenizer and tests
|
2019-06-21 11:09:51 +02:00 |
|
thomwolf
|
45709d7532
|
model running with simple inputs
|
2019-06-21 00:28:42 +02:00 |
|
thomwolf
|
b407972e27
|
update gitignore
|
2019-06-20 13:52:56 +02:00 |
|
thomwolf
|
c2ea5aef77
|
work in progress on xlnet
|
2019-06-20 13:52:21 +02:00 |
|
thomwolf
|
de713fa9b4
|
starting
|
2019-06-20 10:54:19 +02:00 |
|
thomwolf
|
c304593d8f
|
BERTology details in readme
|
2019-06-20 10:05:06 +02:00 |
|
thomwolf
|
411981a080
|
remove slow circle-ci
|
2019-06-20 08:54:18 +02:00 |
|
thomwolf
|
edfe91c36e
|
first version bertology ok
|
2019-06-19 23:43:04 +02:00 |
|
thomwolf
|
7766ce66dd
|
update bertology
|
2019-06-19 22:29:51 +02:00 |
|
thomwolf
|
7f00a36e27
|
pruning should keep on device
|
2019-06-19 22:23:12 +02:00 |
|
thomwolf
|
e4b46d86ce
|
update head pruning
|
2019-06-19 22:16:30 +02:00 |
|
thomwolf
|
0f40e8d6a6
|
debugger
|
2019-06-19 15:38:46 +02:00 |
|
thomwolf
|
0e1e8128bf
|
more logging
|
2019-06-19 15:35:49 +02:00 |
|
thomwolf
|
909d4f1af2
|
cuda again
|
2019-06-19 15:32:10 +02:00 |
|
thomwolf
|
14f0e8e557
|
fix cuda
|
2019-06-19 15:29:28 +02:00 |
|
thomwolf
|
34d706a0e1
|
pruning in bertology
|
2019-06-19 15:25:49 +02:00 |
|
thomwolf
|
dc8e0019b7
|
updating examples
|
2019-06-19 13:23:20 +02:00 |
|
thomwolf
|
68ab9599ce
|
small fix and updates to readme
|
2019-06-19 09:38:38 +02:00 |
|
thomwolf
|
f7e2ac01ea
|
update barrier
|
2019-06-18 22:43:35 +02:00 |
|
thomwolf
|
4d8c4337ae
|
test barrier in distrib training
|
2019-06-18 22:41:28 +02:00 |
|
thomwolf
|
3359955622
|
updating run_classif
|
2019-06-18 22:23:10 +02:00 |
|
thomwolf
|
29b7b30eaa
|
updating evaluation on a single gpu
|
2019-06-18 22:20:21 +02:00 |
|
thomwolf
|
7d2001aa44
|
overwrite_output_dir
|
2019-06-18 22:13:30 +02:00 |
|
thomwolf
|
16a1f338c4
|
fixing
|
2019-06-18 17:06:31 +02:00 |
|
thomwolf
|
92e0ad5aba
|
no numpy
|
2019-06-18 17:00:52 +02:00 |
|
thomwolf
|
4e6edc3274
|
hop
|
2019-06-18 16:57:15 +02:00 |
|
thomwolf
|
f55b60b9ee
|
fixing again
|
2019-06-18 16:56:52 +02:00 |
|
thomwolf
|
8bd9118294
|
quick fix
|
2019-06-18 16:54:41 +02:00 |
|
thomwolf
|
3e847449ad
|
fix out_label_ids
|
2019-06-18 16:53:31 +02:00 |
|
thomwolf
|
aad3a54e9c
|
fix paths
|
2019-06-18 16:48:04 +02:00 |
|
thomwolf
|
40dbda6871
|
updating classification example
|
2019-06-18 16:45:52 +02:00 |
|
thomwolf
|
7388c83b60
|
update run_classifier for distributed eval
|
2019-06-18 16:32:49 +02:00 |
|
thomwolf
|
9727723243
|
fix pickle
|
2019-06-18 16:02:42 +02:00 |
|
thomwolf
|
9710b68dbc
|
fix pickles
|
2019-06-18 16:01:15 +02:00 |
|
thomwolf
|
15ebd67d4e
|
cache in run_classifier + various fixes to the examples
|
2019-06-18 15:58:22 +02:00 |
|
thomwolf
|
e6e5f19257
|
fix
|
2019-06-18 14:45:14 +02:00 |
|
thomwolf
|
a432b3d466
|
distributed traing t_total
|
2019-06-18 14:39:09 +02:00 |
|
thomwolf
|
c5407f343f
|
split squad example in two
|
2019-06-18 14:29:03 +02:00 |
|
thomwolf
|
335f57baf8
|
only on main process
|
2019-06-18 14:03:46 +02:00 |
|
thomwolf
|
326944d627
|
add tensorboard to run_squad
|
2019-06-18 14:02:42 +02:00 |
|
thomwolf
|
d82e5deeb1
|
set find_unused_parameters=True in DDP
|
2019-06-18 12:13:14 +02:00 |
|
thomwolf
|
a59abedfb5
|
DDP update
|
2019-06-18 12:06:26 +02:00 |
|
thomwolf
|
2ef5e0de87
|
switch to pytorch DistributedDataParallel
|
2019-06-18 12:03:13 +02:00 |
|
thomwolf
|
9ce37af99b
|
oups
|
2019-06-18 11:47:54 +02:00 |
|
thomwolf
|
a40955f071
|
no need to duplicate models anymore
|
2019-06-18 11:46:14 +02:00 |
|
thomwolf
|
f964753090
|
explanation on the current location of the caching folder
|
2019-06-18 11:36:28 +02:00 |
|
thomwolf
|
868de8d1d7
|
updating weights loading
|
2019-06-18 10:58:20 +02:00 |
|
thomwolf
|
64e0adda81
|
better error message
|
2019-06-18 10:51:31 +02:00 |
|
thomwolf
|
382e2d1e50
|
spliting config and weight files for bert also
|
2019-06-18 10:37:16 +02:00 |
|
thomwolf
|
4447f270b2
|
updating hub
|
2019-06-17 16:21:28 +02:00 |
|
thomwolf
|
33d3db5c43
|
updating head masking, readme and docstrings
|
2019-06-17 15:51:28 +02:00 |
|
thomwolf
|
965f172de6
|
output all hidden layers states in GPT/GPT-2
|
2019-06-17 14:34:12 +02:00 |
|
thomwolf
|
f12007e421
|
add head masking and pruning to openai GPT
|
2019-06-17 14:19:40 +02:00 |
|
thomwolf
|
b860e47cf5
|
add head masking and pruning to gpt-2
|
2019-06-17 14:12:10 +02:00 |
|
thomwolf
|
7220d47a1c
|
adding head pruning and tests
|
2019-06-17 13:20:45 +02:00 |
|
thomwolf
|
8415a38b23
|
better error messages
|
2019-06-17 13:03:48 +02:00 |
|
thomwolf
|
96c4d3d988
|
add head masking tests
|
2019-06-17 12:17:26 +02:00 |
|
thomwolf
|
34858ae1d9
|
adding bert whole words, bertgerman and gpt-2 medium models, head masking
|
2019-06-17 11:02:39 +02:00 |
|
thomwolf
|
44e9ddd7fe
|
fix num_special_tokens in GPT 2 test
|
2019-06-14 17:17:43 +02:00 |
|
thomwolf
|
5e1207b8ad
|
add attention to all bert models and add test
|
2019-06-14 16:28:25 +02:00 |
|
thomwolf
|
bcc9e93e6f
|
fix test
|
2019-06-14 15:38:20 +02:00 |
|
thomwolf
|
a3274ac40b
|
adding attention outputs in bert
|
2019-06-03 16:11:45 -05:00 |
|
thomwolf
|
275179a003
|
output attentions in GPT-2
|
2019-05-08 22:24:42 +02:00 |
|
thomwolf
|
366a3b0285
|
clean up in tokenization
|
2019-05-08 21:43:51 +02:00 |
|
thomwolf
|
0efc4ab632
|
adding dropout to GPT-2 and embedding dropout to GPT
|
2019-05-08 10:41:35 +02:00 |
|
thomwolf
|
ea9dbea9d5
|
update GPT2 loss computation for more flexbility
|
2019-05-07 23:27:18 +02:00 |
|
thomwolf
|
ce86336545
|
add predict_special_tokens option to GPT also
|
2019-05-07 16:47:22 +02:00 |
|