Commit Graph
100 Commits
Author SHA1 Message Date
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
thomwolf d1b6979aa5 GPT-2 option to avoid predicting special tokens 2019-05-07 16:25:53 +02:00
thomwolf e211785ada extract attention weights from GPT 2019-05-02 18:31:26 +02:00
thomwolf db98a4a48b gpt-2 tokenizer 2019-05-01 11:40:48 +02:00
thomwolf cd110835a0 coverage in circle-ci 2019-04-30 11:35:40 +02:00
thomwolf 80f53f7380 gpt-2 from_pretrained can use special tokens 2019-04-30 11:10:22 +02:00
thomwolf e79ceb1533 gpt-2 special tokens 2019-04-30 11:05:54 +02:00
thomwolf 1f5fc95b68 add code coverage 2019-04-30 11:05:26 +02:00
thomwolf c30139a013 add special tokens to gpt-2 2019-04-30 10:45:26 +02:00
thomwolf b832d5bb8a Release: 0.6.2 2019-04-25 21:37:47 +02:00
thomwolf 80f995a141 revert BertForMultipleChoice linear classifier 2019-04-24 16:51:54 +02:00
thomwolf d94c6b0144 fix training schedules in examples to match new API 2019-04-23 11:17:06 +02:00
thomwolf 34ae5bf838 small clean up in tests 2019-04-17 14:52:12 +02:00
thomwolf 23d4554ec0 is python 2 happy now 2019-04-17 14:48:34 +02:00
thomwolf 265550ec34 relax network connection requirements 2019-04-17 14:22:35 +02:00
thomwolf fa76520240 fix file_utils on python 2 2019-04-17 13:32:22 +02:00
thomwolf bcde2c61cb fix #497 2019-04-17 12:35:38 +02:00
thomwolf 929579f3b5 fix #497 2019-04-17 12:35:08 +02:00
thomwolf 31d387604c adding s3 model tests with --runslow 2019-04-17 11:58:27 +02:00
thomwolf 5afa497cbf fix GPT-2 tokenization to work also on python 3... 2019-04-17 11:04:41 +02:00
thomwolf bc70779bf0 fixed GPT-2 tokenization on python 2 2019-04-17 10:56:15 +02:00