Update convert.py

convert.py

@@ -1,4 +1,5 @@
 # Copied from https://github.com/nghuyong/ERNIE-Pytorch/blob/master/convert.py
+# with some modifications for ernie-m
 
 
 #!/usr/bin/env python
@@ -99,19 +100,8 @@ def extract_and_convert(input_dir, output_dir):
         paddle_paddle_params, _ = D.load_dygraph(os.path.join(input_dir, 'model_state.pdparams'))
     for weight_name, weight_value in paddle_paddle_params.items():
         if 'weight' in weight_name:
-            # if 'encoder' in weight_name or 'pooler' in weight_name or 'cls.' in weight_name:
-            #     weight_value = weight_value.transpose()
-
-            # if 'encoder' in weight_name or 'pooler' in weight_name or 'cls.' in weight_name and \
-            #         "k_proj" not in weight_name and "v_proj" not in weight_name and \
-            #         "out_proj" not in weight_name and "linear1" not in weight_name and \
-            #         "linear2" not in weight_name:
-            #     weight_value = weight_value.transpose()
-            if "encoder" in weight_name:
-                if "linear1" in weight_name or "linear2" in weight_name:
-                    weight_value = weight_value.transpose()
-                else:
-                    weight_value = weight_value.transpose()
+            if 'encoder' in weight_name or 'pooler' in weight_name or 'cls.' in weight_name:
+                weight_value = weight_value.transpose()
 
         if weight_name not in weight_map:
             print('=' * 20, '[SKIP]', weight_name, '=' * 20)