fix

Author: hiworldwzj

test/model/test_model.py

@@ -23,9 +23,9 @@ def test_model_infer(self):
     import torch
 
     parser = make_argument_parser()
-    parser.add_argument("--batch_size", type=int, default=200, help="batch size")
-    parser.add_argument("--input_len", type=int, default=1000, help="input sequence length")
-    parser.add_argument("--output_len", type=int, default=2000, help="output sequence length")
+    parser.add_argument("--batch_size", type=int, default=2, help="batch size")
+    parser.add_argument("--input_len", type=int, default=4096, help="input sequence length")
+    parser.add_argument("--output_len", type=int, default=128, help="output sequence length")
     parser.add_argument(
         "--profile",
         action="store_true",