Commit a3dec59c authored by 王敏's avatar 王敏
Browse files

优化benchmark moe

parent 44d451ff
...@@ -641,7 +641,7 @@ if __name__ == "__main__": ...@@ -641,7 +641,7 @@ if __name__ == "__main__":
parser.add_argument("--seed", type=int, default=0) parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--batch-size", type=int, required=False) parser.add_argument("--batch-size", type=int, required=False)
parser.add_argument("--tune", action="store_true") parser.add_argument("--tune", action="store_true")
parser.add_argument("--nn_moe", type=bool, default=True) parser.add_argument("--nn-moe", action='store_true', default=False)
parser.add_argument("--trust-remote-code", action="store_true") parser.add_argument("--trust-remote-code", action="store_true")
parser.add_argument("--moe-ep-size", type=int, default=1) parser.add_argument("--moe-ep-size", type=int, default=1)
parser.add_argument("--num-gpus", type=int, default=1) parser.add_argument("--num-gpus", type=int, default=1)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment