ConvBERT fix torch <> tf weights conversion (#10314)

* convbert conversion test * fin * fin * fin * clean up tf<->pt conversion * remove from_pt Co-authored-by: patrickvonplaten <patrick.v.platen@gmail.com>
2021-02-24 12:55:34 +01:00
parent cd48078ce5
commit 0d4c9808c4
4 changed files with 11 additions and 6 deletions
--- a/tests/test_modeling_tf_convbert.py
+++ b/tests/test_modeling_tf_convbert.py
@@ -384,8 +384,6 @@ class TFConvBertModelIntegrationTest(unittest.TestCase):
        expected_shape = [1, 6, 768]
        self.assertEqual(output.shape, expected_shape)

-        print(output[:, :3, :3])
-
        expected_slice = tf.constant(
            [
                [