LysandreJik
87747518e9
Blocks deletion from already deleted heads. Necessary integration test.
...
Now raises a warning when a head to be deleted already has been deleted. An integration test verifying the total pipeline (-> from config -> save model -> load model -> additional head pruning) has been added.
2019-08-31 00:33:50 -04:00
LysandreJik
719cb3738d
Pruning for GPT and GPT-2
2019-08-31 00:33:50 -04:00
LysandreJik
fc1fbae45d
XLM can be pruned
2019-08-31 00:33:50 -04:00
Lysandre
42e00cf9e1
Pruning saved to configuration first try
2019-08-31 00:33:50 -04:00
LysandreJik
7f006cdd87
Set seed for head_masking test
2019-08-30 23:58:49 -04:00
Thomas Wolf
50e615f43d
Merge branch 'master' into improved_testing
2019-08-30 13:40:35 +02:00
LysandreJik
c513415b19
Dilbert tests from CommonTests
2019-08-27 23:59:00 -04:00
Lysandre
47267ba556
OpenAI GPT-2 now depends on CommonTests.
2019-08-21 17:50:16 -04:00
Wei-Sheng Chin
c4e9615691
Fix a path so that test can run on Windows
2019-07-17 09:08:40 -07:00
thomwolf
2918b7d2a0
updating tests
2019-07-12 10:57:58 +02:00
LysandreJik
3fbceed8d2
Fix layer reference loss + previous attempted fix
2019-07-11 22:29:55 -04:00
LysandreJik
6c2ee16c04
Test suite testing the tie_weights function as well as the resize_token_embeddings function.
...
Patched an issue relating to the tied weights I had introduced with the TorchScript addition.
Byte order mark management in TSV glue reading.
2019-07-11 22:09:16 -04:00
thomwolf
bd404735a7
embeddings resizing + tie_weights
2019-07-12 00:02:49 +02:00