Fix input_mask shape in inputs (#3992)

PaddleNLP/PaddleLARK/BERT/model/classifier.py

@@ -25,8 +25,8 @@ from model.bert import BertModel
 def create_model(args, bert_config, num_labels, is_prediction=False):
     input_fields = {
         'names': ['src_ids', 'pos_ids', 'sent_ids', 'input_mask', 'labels'],
-        'shapes': [[None, None], [None, None], [None, None],
-                   [None, args.max_seq_len, 1], [None, 1]],
+        'shapes': [[None, None], [None, None], [None, None], [None, None, 1],
+                   [None, 1]],
         'dtypes': ['int64', 'int64', 'int64', 'float32', 'int64'],
         'lod_levels': [0, 0, 0, 0, 0],
     }

PaddleNLP/PaddleLARK/BERT/run_squad.py

@@ -111,7 +111,7 @@ def create_model(bert_config, is_training=False):
         input_fields = {
             'names': ['src_ids', 'pos_ids', 'sent_ids', 'input_mask', 'start_positions', 'end_positions'],
             'shapes': [[None, None], [None, None], [None, None],
-                    [None, args.max_seq_len, 1], [None, 1], [None, 1]],
+                    [None, None, 1], [None, 1], [None, 1]],
             'dtypes': [
                 'int64', 'int64', 'int64', 'float32', 'int64', 'int64'],
             'lod_levels': [0, 0, 0, 0, 0, 0],
@@ -120,7 +120,7 @@ def create_model(bert_config, is_training=False):
         input_fields = {
             'names': ['src_ids', 'pos_ids', 'sent_ids', 'input_mask', 'unique_id'],
             'shapes': [[None, None], [None, None], [None, None],
-                    [None, args.max_seq_len, 1], [None, 1]],
+                    [None, None, 1], [None, 1]],
             'dtypes': [
                 'int64', 'int64', 'int64', 'float32', 'int64'],
             'lod_levels': [0, 0, 0, 0, 0],