github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-01 09:07:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

[bug] fix get_default_parser in examples (#4764)

Browse Source
This commit is contained in:
Baizhou Zhang
2023-09-21 10:42:25 +08:00
committed by GitHub
parent c0a033700c
commit df66741f77
14 changed files with 36 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/language/gpt/titans/train_gpt.py
View File

@@ -1,3 +1,4 @@
import argparse
import contextlib
import os
@@ -29,7 +30,7 @@ VOCAB_SIZE = 50257
def main():
parser = colossalai.get_default_parser()
parser = argparse.ArgumentParser()
parser.add_argument("--from_torch", default=False, action="store_true")
parser.add_argument("--use_dummy_dataset", default=False, action="store_true")
args = parser.parse_args()