LysandreJik
|
4703148f0c
|
TransformerXL can't be exported to TorchScript because of control-flow. Exception added to tests.
|
2019-07-03 14:50:23 -04:00 |
LysandreJik
|
971c24687f
|
XLNET can be exported to TorchScript
|
2019-07-03 11:03:09 -04:00 |
LysandreJik
|
be54b16960
|
GPT can be exported to TorchScript
|
2019-07-02 18:09:45 -04:00 |
LysandreJik
|
d8e83de792
|
GPT2 can be exported to TorchScript
|
2019-07-02 18:01:09 -04:00 |
LysandreJik
|
e891bb43d5
|
BERT can be exported to TorchScript
|
2019-07-02 17:23:18 -04:00 |
LysandreJik
|
6ce1ee04fc
|
TorchScript testing with output_attentions and output_hidden_state
|
2019-07-02 17:22:59 -04:00 |
thomwolf
|
7ed5bf706f
|
add tests
|
2019-07-02 16:42:22 +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 |
Thomas Wolf
|
12e892e174
|
Merge pull request #697 from huggingface/updating_examples
Updating examples
|
2019-06-20 09:58:24 +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 |