XLM can be pruned

2019-08-21 18:57:30 -04:00
parent 42e00cf9e1
commit fc1fbae45d
2 changed files with 6 additions and 1 deletions
--- a/pytorch_transformers/tests/modeling_common_test.py
+++ b/pytorch_transformers/tests/modeling_common_test.py
@@ -269,7 +269,6 @@ class CommonTestCases:
                shutil.rmtree(directory)

        def test_head_pruning_save_load_from_config_init(self):
-            print(self.test_pruning)
            if not self.test_pruning:
                return